Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
update single gpu script
  • Loading branch information
brettkoonce committed Apr 20, 2018
1 parent 0031f89 commit a2ff907
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cifar10/dawn_submission/README.md
Expand Up @@ -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.