Permalink
Browse files

update single gpu script

  • Loading branch information...
brettkoonce committed Apr 20, 2018
1 parent 0031f89 commit a2ff907cd5e7315088324e1ff324fd25a1a67fe3
Showing with 1 addition and 1 deletion.
  1. +1 −1 cifar10/dawn_submission/README.md
@@ -21,4 +21,4 @@ Dataset from:
In the directory on paperspace using a volta (same setup as above), run:
python dawn_mod.py ~/data/cifar10/ --save-dir ~/data/cf_train_save/wrn_small -a wrn_22 --loss-scale 1 --epochs 1 --cycle-len 30 --lr 1.5 --wd 1e-4 --use-clr 20,20,0.95,0.85
python dawn_mod.py ~/data/cifar10/ --save-dir ~/data/cf_train_save/wrn_v5 -a wrn_22 --fp16 --loss-scale 512 --epochs 1 --cycle-len 30 --lr 1.5 --wd 1e-4 --use-clr 20,20,0.95,0.85

0 comments on commit a2ff907

Please sign in to comment.