Created
August 20, 2015 07:53
-
-
Save jameslyons/2e4a7d1caec671e54ee5 to your computer and use it in GitHub Desktop.
caffe solver for iris dataset
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# The train/test net protocol buffer definition | |
net: "./iris_caffe_network.prototxt" | |
# test_iter specifies how many forward passes the test should carry out. | |
# In the case of iris, we have test batch size 10 (specified in iris_network.prototxt) | |
# and 5 test iterations covering the full 50 test vectors. | |
test_iter: 5 | |
# Carry out testing every 500 training iterations. | |
test_interval: 100 | |
# The base learning rate, momentum and the weight decay of the network. | |
base_lr: 0.001 | |
momentum: 0.9 | |
weight_decay: 0.0005 | |
# The learning rate policy | |
lr_policy: "inv" | |
gamma: 0.0001 | |
power: 0.75 | |
# The maximum number of iterations | |
max_iter: 1000 | |
# snapshot intermediate results | |
snapshot: 1000 | |
snapshot_prefix: "./" | |
# solver mode: CPU or GPU | |
solver_mode: GPU |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment