first commit

Zheng222 · May 2, 2018 · 80a5309 · 80a5309
1 parent 84c9001
commit 80a5309
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/train/train.sh b/train/train.sh
@@ -1,3 +1,6 @@
 #!/usr/bin/env bash
+# train from scratch
+~/local/caffe/build/tools/caffe train -solver IDN_solver.prototxt -gpu 0 2>&1 | tee -a IDN_x2.log
 
-~/local/caffe/build/tools/caffe train -solver IDN_solver.prototxt -gpu 0 2>&1 | tee -a IDN_x2.log
+# train from pretrained model
+#~/local/caffe/build/tools/caffe train -solver IDN_solver.prototxt -weights ../test/caffemodel/IDN_x2.caffemodel -gpu 0 2>&1 | tee -a IDN_x2.log