Added prints to matcaffe_demo stages

oceanfire · Mar 31, 2014 · e3daae6 · e3daae6
1 parent 372fadd
commit e3daae6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/matlab/caffe/matcaffe_demo.m b/matlab/caffe/matcaffe_demo.m
@@ -64,18 +64,24 @@
     % NOTE: you'll have to get network definition
     error('You need the network prototxt definition');
   end
-  caffe('init', model_def_file, model_file);
+  caffe('init', model_def_file, model_file)
 end
 
+fprintf('Done with init\n');
+
 % set to use GPU or CPU
 if exist('use_gpu', 'var') && use_gpu
+  fprintf('Using GPU Mode\n');
   caffe('set_mode_gpu');
 else
+  fprintf('Using CPU Mode\n');
   caffe('set_mode_cpu');
 end
 
+fprintf('Done with set_mode\n');
 % put into test mode
 caffe('set_phase_test');
+fprintf('Done with set_phase_test\n');
 
 % prepare oversampled input
 % input_data is Height x Width x Channel x Num