cnn_train.m: fixes a bug in computing the errors when using more than…

… one sub batch
dbbert · Apr 28, 2015 · 1db6764 · 1db6764
1 parent 80ef8a5
commit 1db6764
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/cnn_train.m b/examples/cnn_train.m
@@ -250,6 +250,7 @@
   batchSize = min(opts.batchSize, numel(subset) - t + 1) ;
   batchTime = tic ;
   numDone = 0 ;
+  error = [] ;
   for s=1:opts.numSubBatches
     % get this image batch and prefetch the next
     batchStart = t + (labindex-1) + (s-1) * numlabs ;
@@ -283,9 +284,9 @@
                       'sync', opts.sync) ;
 
     % accumulate training errors
-    error = [...
+    error = sum([error, [...
       sum(double(gather(res(end).x))) ;
-      reshape(opts.errorFunction(opts, labels, res),[],1) ; ] ;
+      reshape(opts.errorFunction(opts, labels, res),[],1) ; ]],2) ;
     numDone = numDone + numel(batch) ;
   end