diff --git a/ml/cmsisnn/models/cifar10/train.sh b/ml/cmsisnn/models/cifar10/train.sh index 2b6ae0b8c..93549ea08 100755 --- a/ml/cmsisnn/models/cifar10/train.sh +++ b/ml/cmsisnn/models/cifar10/train.sh @@ -1,17 +1,18 @@ #!/usr/bin/env sh set -e +DIR=models/cifar10 TOOLS=./caffe/build/tools $TOOLS/caffe train \ - --solver=models/cifar10/cifar10_solver.prototxt $@ + --solver=models/cifar10/cifar10_solver.prototxt $@ 2>&1 | tee $DIR/training.log # reduce learning rate by factor of 10 $TOOLS/caffe train \ --solver=models/cifar10/cifar10_solver_lr1.prototxt \ - --snapshot=models/cifar10/cifar10_iter_60000.solverstate.h5 $@ + --snapshot=models/cifar10/cifar10_iter_60000.solverstate.h5 $@ 2>&1 | tee -a $DIR/training.log # reduce learning rate by factor of 10 $TOOLS/caffe train \ --solver=models/cifar10/cifar10_solver_lr2.prototxt \ - --snapshot=models/cifar10/cifar10_iter_65000.solverstate.h5 $@ + --snapshot=models/cifar10/cifar10_iter_65000.solverstate.h5 $@ 2>&1 | tee -a $DIR/training.log diff --git a/ml/cmsisnn/models/cifar10_fast/train.sh b/ml/cmsisnn/models/cifar10_fast/train.sh index f3508aa73..f884ec06b 100755 --- a/ml/cmsisnn/models/cifar10_fast/train.sh +++ b/ml/cmsisnn/models/cifar10_fast/train.sh @@ -1,17 +1,18 @@ #!/usr/bin/env sh set -e +DIR=models/cifar10_fast TOOLS=./caffe/build/tools $TOOLS/caffe train \ - --solver=models/cifar10_fast/cifar10_fast_solver.prototxt $@ + --solver=models/cifar10_fast/cifar10_fast_solver.prototxt $@ 2>&1 | tee $DIR/training.log # reduce learning rate by factor of 10 $TOOLS/caffe train \ --solver=models/cifar10_fast/cifar10_fast_solver_lr1.prototxt \ - --snapshot=models/cifar10_fast/cifar10_fast_iter_60000.solverstate.h5 $@ + --snapshot=models/cifar10_fast/cifar10_fast_iter_60000.solverstate.h5 $@ 2>&1 | tee -a $DIR/training.log # reduce learning rate by factor of 10 $TOOLS/caffe train \ --solver=models/cifar10_fast/cifar10_fast_solver_lr2.prototxt \ - --snapshot=models/cifar10_fast/cifar10_fast_iter_65000.solverstate.h5 $@ + --snapshot=models/cifar10_fast/cifar10_fast_iter_65000.solverstate.h5 $@ 2>&1 | tee -a $DIR/training.log diff --git a/ml/cmsisnn/models/lenet/train.sh b/ml/cmsisnn/models/lenet/train.sh index 8af4b0fe2..17c26bb49 100755 --- a/ml/cmsisnn/models/lenet/train.sh +++ b/ml/cmsisnn/models/lenet/train.sh @@ -1,7 +1,8 @@ #!/usr/bin/env sh set -e +DIR=models/lenet TOOLS=./caffe/build/tools $TOOLS/caffe train \ - --solver=models/lenet/lenet_solver.prototxt $@ + --solver=models/lenet/lenet_solver.prototxt $@ 2>&1 | tee $DIR/training.log diff --git a/ml/cmsisnn/models/smile/train.sh b/ml/cmsisnn/models/smile/train.sh index 4f69ddb14..5f86e41d1 100755 --- a/ml/cmsisnn/models/smile/train.sh +++ b/ml/cmsisnn/models/smile/train.sh @@ -1,7 +1,8 @@ #!/usr/bin/env sh set -e +DIR=models/smile TOOLS=./caffe/build/tools $TOOLS/caffe train \ - --solver=models/smile/smile_solver.prototxt $@ + --solver=models/smile/smile_solver.prototxt $@ 2>&1 | tee $DIR/training.log