Skip to content
Snippets Groups Projects
Commit d846104c authored by Franck Dary's avatar Franck Dary
Browse files

Allow batch size 1

parent 9f9a9f9d
No related branches found
No related tags found
No related merge requests found
......@@ -82,6 +82,8 @@ float Trainer::epoch(bool printAdvancement)
auto prediction = machine.getClassifier()->getNN()(data);
labels = labels.reshape(labels.dim() == 0 ? 1 : labels.size(0));
auto loss = lossFct(prediction, labels);
try
{
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment