Permalink
Browse files

add v2 multi-gpu script

  • Loading branch information...
brettkoonce committed Apr 20, 2018
1 parent a2ff907 commit 1bc5aeec765572397c0ebe4a5d616d03beeeeec1
Showing with 6 additions and 5 deletions.
  1. +6 −5 cifar10/dawn_submission/README.md
@@ -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).

0 comments on commit 1bc5aee

Please sign in to comment.