From 0aa5de476e35442dd9be0d29e2f848fa594fd7bb Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Mon, 5 Aug 2019 14:30:02 -0700 Subject: [PATCH 1/2] Initial version of README for classification reference scripts --- references/classification/README.md | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 references/classification/README.md diff --git a/references/classification/README.md b/references/classification/README.md new file mode 100644 index 00000000000..ca568650a2c --- /dev/null +++ b/references/classification/README.md @@ -0,0 +1,25 @@ +# Image classification reference training scripts + + +### ResNext-50 32x4d +``` +python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\ + --model resnext50_32x4d --epochs 100 +``` + + +### ResNext-101 32x8d + +On 8 nodes, each with 8 GPUs (for a total of 64 GPUS) +``` +python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\ + --model resnext101_32x8d --epochs 100 +``` + + +### MobileNetV2 +``` +python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\ + --model mobilenet_v2 --epochs 300 --lr 0.045 --wd 0.00004\ + --lr-step-size 1 --lr-gamma 0.98 +``` From ce3df33b550fbfc730f45ae83f53bf289a80190a Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Mon, 5 Aug 2019 15:21:22 -0700 Subject: [PATCH 2/2] More context --- references/classification/README.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/references/classification/README.md b/references/classification/README.md index ca568650a2c..da4477b8304 100644 --- a/references/classification/README.md +++ b/references/classification/README.md @@ -1,5 +1,10 @@ # Image classification reference training scripts +This folder contains reference training scripts for image classification. +They serve as a log of how to train specific models, as provide baseline +training and evaluation scripts to quickly bootstrap research. + +Except otherwise noted, all models have been trained on 8x V100 GPUs. ### ResNext-50 32x4d ```