--- a +++ b/solver.prototxt @@ -0,0 +1,16 @@ +train_net: "train_3d_denseseg.prototxt" +base_lr: 0.0002 +display: 20 +max_iter: 200000 +lr_policy: "step" +gamma: 0.1 +momentum: 0.97 +weight_decay: 0.0005 +stepsize: 50000 +snapshot: 2000 +snapshot_prefix: "./snapshot/3d_denseseg_iseg" +solver_mode: GPU +random_seed: 831486 +average_loss: 20 +iter_size: 1 +type: "Adam"