Skip to content

GitLab

"...text-generation-inference.git" did not exist on "19c41824cb11ba1a3b60a2a65274d8c074383de3"
Unverified Commit 40555b3a authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

Update README.md

parent a74a0f20
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/imagenet/README.md
View file @ 40555b3a
...@@ -58,7 +58,7 @@ $ python main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 ./ ...@@ -58,7 +58,7 @@ $ python main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 ./
$ python main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 --loss-scale 128.0 ./ $ python main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 --loss-scale 128.0 ./
$ python -m torch.distributed.launch --nproc_per_node=2 main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 ./ $ python -m torch.distributed.launch --nproc_per_node=2 main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 ./
``` ```
Options are explained below. Options are explained below. Again, the [updated API guide](https://nvidia.github.io/apex/amp.html) provides more detail.
#### `--opt-level O0` (FP32 training) and `O3` (FP16 training) #### `--opt-level O0` (FP32 training) and `O3` (FP16 training)
......
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