Enable logging in NN scripts.

This commit is contained in:
iabdalkader 2019-03-29 22:33:09 +02:00
parent 5fc022718b
commit db6a55ace3
4 changed files with 12 additions and 8 deletions

View File

@ -1,17 +1,18 @@
#!/usr/bin/env sh #!/usr/bin/env sh
set -e set -e
DIR=models/cifar10
TOOLS=./caffe/build/tools TOOLS=./caffe/build/tools
$TOOLS/caffe train \ $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 # reduce learning rate by factor of 10
$TOOLS/caffe train \ $TOOLS/caffe train \
--solver=models/cifar10/cifar10_solver_lr1.prototxt \ --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 # reduce learning rate by factor of 10
$TOOLS/caffe train \ $TOOLS/caffe train \
--solver=models/cifar10/cifar10_solver_lr2.prototxt \ --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

View File

@ -1,17 +1,18 @@
#!/usr/bin/env sh #!/usr/bin/env sh
set -e set -e
DIR=models/cifar10_fast
TOOLS=./caffe/build/tools TOOLS=./caffe/build/tools
$TOOLS/caffe train \ $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 # reduce learning rate by factor of 10
$TOOLS/caffe train \ $TOOLS/caffe train \
--solver=models/cifar10_fast/cifar10_fast_solver_lr1.prototxt \ --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 # reduce learning rate by factor of 10
$TOOLS/caffe train \ $TOOLS/caffe train \
--solver=models/cifar10_fast/cifar10_fast_solver_lr2.prototxt \ --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

View File

@ -1,7 +1,8 @@
#!/usr/bin/env sh #!/usr/bin/env sh
set -e set -e
DIR=models/lenet
TOOLS=./caffe/build/tools TOOLS=./caffe/build/tools
$TOOLS/caffe train \ $TOOLS/caffe train \
--solver=models/lenet/lenet_solver.prototxt $@ --solver=models/lenet/lenet_solver.prototxt $@ 2>&1 | tee $DIR/training.log

View File

@ -1,7 +1,8 @@
#!/usr/bin/env sh #!/usr/bin/env sh
set -e set -e
DIR=models/smile
TOOLS=./caffe/build/tools TOOLS=./caffe/build/tools
$TOOLS/caffe train \ $TOOLS/caffe train \
--solver=models/smile/smile_solver.prototxt $@ --solver=models/smile/smile_solver.prototxt $@ 2>&1 | tee $DIR/training.log