Skip to content

Instantly share code, notes, and snippets.

@revilokeb
Created June 7, 2016 21:05
Show Gist options
  • Save revilokeb/1029518fc55c8a254b4f24dccba74487 to your computer and use it in GitHub Desktop.
Save revilokeb/1029518fc55c8a254b4f24dccba74487 to your computer and use it in GitHub Desktop.
Caffe solver for training wide residual net wrn_28_10
net: "MY_PATH/wrn_28_10_train_val.prototxt"
test_iter: 157
test_interval: 390
test_initialization: false
iter_size: 1
#type: "RMSProp"
#rms_decay: 0.9
#delta: 1.0
type: "Nesterov"
display: 390
average_loss: 390
base_lr: 0.1
lr_policy: "step"
stepsize: 23437
gamma: 0.2
max_iter: 70312
momentum: 0.9
weight_decay: 0.0005
snapshot: 3130
snapshot_prefix: "MY_PATH/wrn_28_10_1sttry_train"
solver_mode: GPU
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment