add models for default_resnet50

intel · Dec 13, 2017 · 7ef8316 · 7ef8316
1 parent a5c7df7
commit 7ef8316
Show file tree

Hide file tree

Showing 2 changed files with 3,344 additions and 0 deletions.
diff --git a/models/intel_optimized_models/multinode/default_resnet50_16nodes/solver.prototxt b/models/intel_optimized_models/multinode/default_resnet50_16nodes/solver.prototxt
@@ -0,0 +1,19 @@
+net: "models/intel_optimized_models/multinode/default_resnet50_16nodes/train_val.prototxt"
+test_iter: 1000
+test_interval: 625
+test_initialization: false
+display: 40
+base_lr: 0.8
+lr_policy: "multistep"
+stepvalue:18750
+stepvalue:37500
+stepvalue:50000
+gamma: 0.1
+max_iter: 62556   # 56300
+warmup_iter: 3125 # 1281167 / 2048 * 5 epochs
+warmup_start_lr: 0.1
+momentum: 0.9
+weight_decay: 0.0001
+snapshot: 6250
+snapshot_prefix: "default_resnet_50_16_nodes"
+solver_mode: CPU