Skip to content

Commit

Permalink
add v2 multi-gpu script
Browse files Browse the repository at this point in the history
  • Loading branch information
brettkoonce committed Apr 20, 2018
1 parent a2ff907 commit 1bc5aee
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions cifar10/dawn_submission/README.md
Original file line number Diff line number Diff line change
@@ -1,15 +1,16 @@
### Script (multi-gpu)
### Script (multi-gpu, v2)

In the parent directory, run:

time python -m multiproc.py dawn_mod.py /home/ubuntu/data/cifar10 \
--save-dir /home/ubuntu/data/cf_train_save/wrn_submission \
--save-dir /home/ubuntu/data/cf_train_save/wrn_v2_submission \
-a wrn_22 \
--epochs 1 --cycle-len 50 \
--epochs 1 --cycle-len 40 \
--fp16 --loss-scale 512 \
-b 128 \
--wd 2e-4 --lr 1.3 \
--use-clr 50,15,0.95,0.85
--wd 2e-4 --lr 2.0 \
--use-clr 50,12.5,0.95,0.85 \
-j 8

on an Amazon p3.16xlarge instance. You will need to symlink (ln -s YOUR_FASTAI_DIRECTORY fastai) your fastai install inside the cifar directory for the python imports (or do git clone https://github.com/fastai/fastai.git inside the parent directory). Uses fastai master (current SHA1: https://github.com/fastai/fastai/commit/f9c3d22cdb79e9438c308e3baeb91f04944081fe) and pytorch master (.4) (https://github.com/pytorch/pytorch/commit/354dac976905ba718ea94a73c11587438ae86cde).

Expand Down

0 comments on commit 1bc5aee

Please sign in to comment.