forked from intel/caffe
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update of Intel oprimized solver for resnet 50
- Loading branch information
1 parent
8956f39
commit 000fb82
Showing
1 changed file
with
14 additions
and
4 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |