Skip to content

Commit

Permalink
Update of Intel oprimized solver for resnet 50
Browse files Browse the repository at this point in the history
  • Loading branch information
pszewc-habana committed Jan 24, 2017
1 parent 8956f39 commit 000fb82
Showing 1 changed file with 14 additions and 4 deletions.
18 changes: 14 additions & 4 deletions models/intel_optimized_models/resnet_50/solver.prototxt
Original file line number Diff line number Diff line change
@@ -1,16 +1,26 @@
#This is Intel(R) optimized (in terms of time to train) version of solver for model described in the [Deep Residual Learning for Image Recognition](https://arxiv.org/pdf/1512.03385v1) publication.
#Original solver.prototxt can be found in /models/default_resnet_50/ directory of this repository.
#Differences:
#- gamma is set to 0.51
#- stepsize is set to 60000
#
#Top-5 and Top-1 results achieved with this version of solver:
#Top-5: 89.16%
#Top-1: 68.86%
#Training was performed using server equipped with Intel(R) Xeon Phi(TM) CPU 7250 processor.
net: "models/intel_optimized_models/resnet_50/train_val.prototxt"
test_iter: 100
test_interval: 10000
test_initialization: false
display: 40
average_loss: 40
base_lr: 0.12
base_lr: 0.1
lr_policy: "step"
stepsize: 40000
gamma: 0.61
stepsize: 60000
gamma: 0.51
max_iter: 600000
momentum: 0.9
weight_decay: 0.0001
snapshot: 40000
snapshot: 50000
snapshot_prefix: "models/intel_optimized_models/resnet_50/resnet_50"
solver_mode: CPU

0 comments on commit 000fb82

Please sign in to comment.