Update runtraining.sh with custom flags to allow caller to adjust learning rate etc.

CRBS / cdeep3m

Please go to https://github.com/CRBS/cdeep3m2 for most recent version

Other

58 stars 10 forks source link

Closed coleslaw481 closed 6 years ago

coleslaw481 commented 6 years ago

Add flags to runtraining.sh to allow user to optionally adjust these values in solver.prototxt:

base_lr: 1e-02
power: 0.8
momentum: 0.9
weight_decay: 0.0005
average_loss: 16
lr_policy: "poly"
iter_size: 8
snapshot: 2000

coleslaw481 commented 6 years ago

Fixed in 0.16.0rc3