added resnet

TheLegendAli · Jul 15, 2016 · 6abd64d · 6abd64d
1 parent 6837095
commit 6abd64d
Show file tree

Hide file tree

Showing 5 changed files with 43,985 additions and 1 deletion.
diff --git a/config/DEEPLABV2-RESNET101/solver.prototxt b/config/DEEPLABV2-RESNET101/solver.prototxt
@@ -0,0 +1,16 @@
+train_net: "${EXP}/config/${NET_ID}/train_${TRAIN_SET}.prototxt"
+
+iter_size: 10
+lr_policy: "poly"
+power: 0.9
+base_lr: 2.5e-4
+
+average_loss: 20
+display: 20
+max_iter: 20000
+momentum: 0.9
+weight_decay: 0.0005
+
+snapshot: 10000
+snapshot_prefix: "${EXP}/model/${NET_ID}/train"
+solver_mode: GPU
diff --git a/config/DEEPLABV2-RESNET101/solver2.prototxt b/config/DEEPLABV2-RESNET101/solver2.prototxt
@@ -0,0 +1,16 @@
+train_net: "${EXP}/config/${NET_ID}/train_${TRAIN_SET}.prototxt"
+
+iter_size: 10
+lr_policy: "poly"
+power: 0.9
+base_lr: 2.5e-4
+
+average_loss: 20
+display: 20
+max_iter: 20000
+momentum: 0.9
+weight_decay: 0.0005
+
+snapshot: 10000
+snapshot_prefix: "${EXP}/model/${NET_ID}/train2"
+solver_mode: GPU