diff --git a/ml/cmsisnn/models/cifar10_fast/cifar10_fast.network b/ml/cmsisnn/models/cifar10_fast/cifar10_fast.network index acbaded90..44e3fc30a 100644 Binary files a/ml/cmsisnn/models/cifar10_fast/cifar10_fast.network and b/ml/cmsisnn/models/cifar10_fast/cifar10_fast.network differ diff --git a/ml/cmsisnn/models/smile/smile.network b/ml/cmsisnn/models/smile/smile.network index e0414f87f..72a045839 100644 Binary files a/ml/cmsisnn/models/smile/smile.network and b/ml/cmsisnn/models/smile/smile.network differ diff --git a/ml/cmsisnn/models/smile/smile_solver.prototxt b/ml/cmsisnn/models/smile/smile_solver.prototxt index 465460ef7..8e49a607f 100644 --- a/ml/cmsisnn/models/smile/smile_solver.prototxt +++ b/ml/cmsisnn/models/smile/smile_solver.prototxt @@ -3,8 +3,8 @@ net: "models/smile/smile_train_test.prototxt" # Specifies how many forward passes the test should carry out. # Test batch size 48*30 iterations = 1440 testing images. test_iter: 30 -# Carry out testing every 500 training iterations. -test_interval: 500 +# Carry out testing every 1000 training iterations. +test_interval: 1000 # The learning rate policy # begin training at a learning rate of 0.01 = 1e-2 base_lr: 0.01 @@ -17,13 +17,14 @@ lr_policy: "step" # (i.e., multiply it by a factor of gamma = 0.1) gamma: 0.1 # Drop the learning rate every 10K iterations -stepsize: 40000 +stepsize: 10000 # train for 100K iterations total -max_iter: 200000 +max_iter: 60000 # Display every 100 iterations display: 100 # snapshot intermediate results snapshot: 20000 +snapshot_format: HDF5 snapshot_prefix: "models/smile/smile" # solver mode: CPU or GPU solver_mode: GPU diff --git a/ml/cmsisnn/models/smile/smile_train_test.prototxt b/ml/cmsisnn/models/smile/smile_train_test.prototxt index 801d063d2..ddb4a43f6 100644 --- a/ml/cmsisnn/models/smile/smile_train_test.prototxt +++ b/ml/cmsisnn/models/smile/smile_train_test.prototxt @@ -59,7 +59,7 @@ layer { } } layer { - name: "dropout" + name: "dropout1" type: "Dropout" bottom: "conv1" top: "conv1" @@ -113,7 +113,7 @@ layer { } } layer { - name: "dropout" + name: "dropout2" type: "Dropout" bottom: "conv2" top: "conv2" @@ -161,7 +161,7 @@ layer { } } layer { - name: "dropout" + name: "dropout3" type: "Dropout" bottom: "conv3" top: "conv3" diff --git a/ml/cmsisnn/models/smile/test.sh b/ml/cmsisnn/models/smile/test.sh index f167c93e7..e00561b8e 100755 --- a/ml/cmsisnn/models/smile/test.sh +++ b/ml/cmsisnn/models/smile/test.sh @@ -5,4 +5,4 @@ TOOLS=./caffe/build/tools $TOOLS/caffe test \ --model=models/smile/smile_train_test.prototxt \ - --weights=models/smile/smile_iter_200000.caffemodel $@ + --weights=models/smile/smile_iter_60000.caffemodel.h5 $@