Added prints to matcaffe_demo stages

diff --git a/matlab/caffe/matcaffe_demo.m b/matlab/caffe/matcaffe_demo.m
index 7296eb1..3021bf0 100644 (file)
--- a/matlab/caffe/matcaffe_demo.m
+++ b/matlab/caffe/matcaffe_demo.m
@@ -64,18 +64,24 @@ if caffe('is_initialized') == 0
     % 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