add prototxt for FB resnet50

intel · Nov 6, 2018 · 987b23b · 987b23b
1 parent ae367fb
commit 987b23b
Show file tree

Hide file tree

Showing 6 changed files with 6,674 additions and 0 deletions.
diff --git a/models/intel_optimized_models/multinode/resnet50_4nodes_4s/README b/models/intel_optimized_models/multinode/resnet50_4nodes_4s/README
@@ -0,0 +1 @@
+-ppn 8
diff --git a/models/intel_optimized_models/multinode/resnet50_4nodes_4s/solver.prototxt b/models/intel_optimized_models/multinode/resnet50_4nodes_4s/solver.prototxt
@@ -0,0 +1,20 @@
+net: "models/intel_optimized_models/mlperf_2018/resnet50_4nodes_4s/train_val.prototxt"
+test_iter: 1000
+test_interval: 1251
+test_initialization: false
+display: 60
+base_lr: 0.4
+lr_policy: "multistep"
+stepvalue:37530 # 30 epochs
+stepvalue:75060 # 60 epochs
+stepvalue:100080 # 80 epochs
+gamma: 0.1
+max_iter: 81315 # 65 epochs
+warmup_iter: 5004 # 1281167 / 1024 * 4 epochs
+warmup_start_lr: 0.1
+momentum: 0.9
+weight_decay: 0.0001
+snapshot: 1251 # 1 epoch
+snapshot_prefix: "resnet50_4nodes_4s_run1"
+solver_mode: CPU
+test_offset: 0