Bug fix

gather in gpu mode
kjw0612 · Sep 19, 2015 · b55e7fb · b55e7fb
1 parent bc4d6b0
commit b55e7fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rcn_train.m b/rcn_train.m
@@ -302,8 +302,8 @@
 
     for problem_iter = 1:numel(opts.problems)
         subplot(2,numel(opts.problems),problem_iter+numel(opts.problems)) ;
-        info.test.error{problem_iter}.base(end+1) = eval_base(problem_iter);
-        info.test.error{problem_iter}.ours(end+1) = eval_ours(problem_iter);
+        info.test.error{problem_iter}.base(end+1) = gather(eval_base(problem_iter));
+        info.test.error{problem_iter}.ours(end+1) = gather(eval_ours(problem_iter));
         plot(1:epoch, info.test.error{problem_iter}.base, 'k') ; hold on ;
         plot(1:epoch, info.test.error{problem_iter}.ours, 'b') ;
         h=legend('Baseline', 'Ours') ;