Skip to content

Instantly share code, notes, and snippets.

@zhreshold
Last active December 7, 2017 22:40
Show Gist options
  • Save zhreshold/89bed4ae69c0d8fa23172a2acea19238 to your computer and use it in GitHub Desktop.
Save zhreshold/89bed4ae69c0d8fa23172a2acea19238 to your computer and use it in GitHub Desktop.
train model zoo
git clone https://github.com/zhreshold/mxnet -b model_zoo
cd mxnet/example/gluon
sudo -H pip install -U mxnet-cu90
python image_classification.py --dataseet --train-data ~/efs/users/joshuazz/data/imagenet/record/train_480_q95.rec --val-data ~/efs/users/joshuazz/data/imagenet/record/val_480_q95.rec --batch-size 64 --num-gpus 4 --epochs 120 --lr 0.1 --mode hybrid --model resnet50_v2 --log-interval 200
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment