Skip to content

GitLab

"...pytorch/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "3774945dce028c2492b1b96d641a58bd7baf2a0f"
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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
references/classification/README.md
View file @ 655a2ede
...@@ -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