Skip to content

Instantly share code, notes, and snippets.

@b0noI
Created February 27, 2018 01:45
Show Gist options
  • Save b0noI/7b41ad0de72efad6f0dcff5da88dcf42 to your computer and use it in GitHub Desktop.
Save b0noI/7b41ad0de72efad6f0dcff5da88dcf42 to your computer and use it in GitHub Desktop.
training execution
#!/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