Unverified Commit 655a2ede authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix training model instructions in references/classification (#4089)

parent b4a075cb
...@@ -23,7 +23,8 @@ Since `AlexNet` and the original `VGG` architectures do not include batch ...@@ -23,7 +23,8 @@ Since `AlexNet` and the original `VGG` architectures do not include batch
normalization, the default initial learning rate `--lr 0.1` is to high. normalization, the default initial learning rate `--lr 0.1` is to high.
``` ```
python main.py --model $MODEL --lr 1e-2 python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
--model $MODEL --lr 1e-2
``` ```
Here `$MODEL` is one of `alexnet`, `vgg11`, `vgg13`, `vgg16` or `vgg19`. Note Here `$MODEL` is one of `alexnet`, `vgg11`, `vgg13`, `vgg16` or `vgg19`. Note
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment