Skip to content
This repository has been archived by the owner on Aug 5, 2022. It is now read-only.

Commit

Permalink
Add benchmarking descriptions into resnet_50 solver prototxt file
Browse files Browse the repository at this point in the history
  • Loading branch information
manofmountain committed Jan 11, 2018
1 parent 69b63e7 commit 187c913
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions models/intel_optimized_models/resnet_50/solver.prototxt
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
#The prototxt files(solver.prototxt and train_val.prototxt) are copied from "models/intel_optimized_models/multinode/resnet_50_16_nodes_2k_batch", temporarily used for throughput benchmarking. It can't achieve expected accuracy for training. We'll provide the files for single node training in the future.
net: "models/intel_optimized_models/resnet_50/train_val.prototxt"
test_iter: 1000
test_interval: 625
Expand All @@ -9,11 +10,9 @@ stepvalue:18750
stepvalue:37500
stepvalue:50000
gamma: 0.1
max_iter: 56300
max_iter: 200
warmup_iter: 3125 # 1281167 / 2048 * 5 epochs
warmup_start_lr: 0.1
momentum: 0.9
weight_decay: 0.0001
snapshot: 6250
snapshot_prefix: "models/intel_optimized_models/resnet_50/resnet_50_"
solver_mode: CPU

0 comments on commit 187c913

Please sign in to comment.