From 839f3292e391d3f9efa58a2cf8fed43fda434bbe Mon Sep 17 00:00:00 2001
From: Shuai Zheng <kylezheng04@gmail.com>
Date: Sun, 10 Jul 2016 21:16:04 +0100
Subject: [PATCH] Update tvg_matcaffe_init.m

update the matlab init model file.
---
 matlab/tvg_matcaffe_init.m | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/matlab/tvg_matcaffe_init.m b/matlab/tvg_matcaffe_init.m
index 83f65e77149..74da27ab2da 100644
--- a/matlab/tvg_matcaffe_init.m
+++ b/matlab/tvg_matcaffe_init.m
@@ -14,7 +14,15 @@
   error('Missing argument model_file');
 end
 
-
+% set to use GPU or CPU
+if use_gpu
+  fprintf('Using GPU Mode\n');
+  caffe.set_mode_gpu();
+  caffe.set_device(gpu_id);
+else
+  fprintf('Using CPU Mode\n');
+  caffe.set_mode_cpu();
+end
 
 if exist(model_file, 'file') ~= 2
     error('You need a network model file');
@@ -27,14 +35,5 @@
 fprintf('Done with set_phase_test\n');
 fprintf('Done with init\n');
 
-% set to use GPU or CPU
-if use_gpu
-  fprintf('Using GPU Mode\n');
-  caffe.set_mode_gpu();
-  caffe.set_device(gpu_id);
-else
-  fprintf('Using CPU Mode\n');
-  caffe.set_mode_cpu();
-end
 fprintf('Done with set_mode\n');