Created
February 27, 2018 01:45
-
-
Save b0noI/7b41ad0de72efad6f0dcff5da88dcf42 to your computer and use it in GitHub Desktop.
training execution
This file contains hidden or 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
#!/bin/bash | |
GPU_TO_TRAIN=0,1,2,3,4,5,6,7 | |
BATCH_SIZE=960 | |
KVSTORE=dist_sync | |
source activate mxnet_p36 | |
python image-classification/train_imagenet.py \ | |
--gpu $GPU_TO_TRAIN --batch-size $BATCH_SIZE --num-epochs 10 \ | |
--data-nthreads 40 --disp-batches 20 \ | |
--network resnet-v1 --num-layers 50 \ | |
--max-random-shear-ratio 0 \ | |
--min-random-scale 0.533 \ | |
--max-random-rotate-angle 0 \ | |
--max-random-h 0 --max-random-l 0 \ | |
--max-random-s 0 --max-random-aspect-ratio 0 \ | |
--kv-store $KVSTORE \ | |
--benchmark 1 & |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment