create_models.py>>predict.prototext fix; keep original caffe path

mitmul · Apr 29, 2015 · 6b025a6 · 6b025a6
1 parent 8fda84c
commit 6b025a6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scripts/create_models.py b/scripts/create_models.py
@@ -607,7 +607,6 @@ def solver(model_name, base_lr, gamma, stepsize, max_iter):
         fp.close()
 
         subprocess.check_output(
-            #['python', '%s/Libraries/caffe/python/draw_net.py' % home_dir,
-            ['python', '%s/Downloads/caffe_dev_mitmul/caffe_ssai/python/draw_net.py' % home_dir,
+            ['python', '%s/Libraries/caffe/python/draw_net.py' % home_dir,
              'models/%s/train_test.prototxt' % model_name,
              'models/%s/net.png' % model_name])