openmv/ml/cmsisnn/models/cifar10/train.sh
2018-10-11 18:19:41 +02:00

18 lines
501 B
Bash
Executable File

#!/usr/bin/env sh
set -e
TOOLS=./caffe/build/tools
$TOOLS/caffe train \
--solver=models/cifar10/cifar10_solver.prototxt $@
# 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 $@
# 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 $@