diff options
Diffstat (limited to 'src/programs/imagenet_solver.prototxt')
| -rw-r--r-- | src/programs/imagenet_solver.prototxt | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/programs/imagenet_solver.prototxt b/src/programs/imagenet_solver.prototxt new file mode 100644 index 00000000..58b0dfef --- /dev/null +++ b/src/programs/imagenet_solver.prototxt @@ -0,0 +1,10 @@ +base_lr: 0.02 +lr_policy: "step" +gamma: 0.1 +stepsize: 340000 +display: 100 +max_iter: 1200000 +momentum: 0.9 +weight_decay: 0.0005 +snapshot: 15000 +snapshot_prefix: "alexnet_train"
\ No newline at end of file |