automatically use the latest snapshot to test

jorgher · Jul 29, 2017 · c80df8b · c80df8b
1 parent 2cf1bc4
commit c80df8b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test.sh b/test.sh
@@ -1,4 +1,9 @@
 #!/bin/sh
+#latest=snapshot/mobilenet_iter_73000.caffemodel
+latest=$(ls -t snapshot/*.caffemodel | head -n 1)
+if test -z $latest; then
+	exit 1
+fi
 ../../build/tools/caffe train -solver="solver_test.prototxt" \
---weights=snapshot/mobilenet_iter_73000.caffemodel \
+--weights=$latest \
 -gpu 0