Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit a74a0f2

Browse files
author
mcarilli
authored
Update README.md
1 parent d2862d5 commit a74a0f2

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

examples/imagenet/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Mixed Precision ImageNet Training in PyTorch
22

33
`main_amp.py` is based on [https://github.com/pytorch/examples/tree/master/imagenet](https://github.com/pytorch/examples/tree/master/imagenet).
4-
It implements Automatic Mixed Precision (Amp) training of popular model architectures, such as ResNet, AlexNet, and VGG, on the ImageNet dataset. Command-line flags forwarded to `amp.initialize` are used easily manipulate and switch between various pure and mixed precision training "optimization levels" or `opt_level`s. For a detailed explanation of `opt_level`s, [refer to the updated API guide](https://nvidia.github.io/apex/amp.html).
4+
It implements Automatic Mixed Precision (Amp) training of popular model architectures, such as ResNet, AlexNet, and VGG, on the ImageNet dataset. Command-line flags forwarded to `amp.initialize` are used to easily manipulate and switch between various pure and mixed precision "optimization levels" or `opt_level`s. For a detailed explanation of `opt_level`s, see the [updated API guide](https://nvidia.github.io/apex/amp.html).
55

66
Three lines enable Amp:
77
```
@@ -58,7 +58,7 @@ $ python main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 ./
5858
$ python main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 --loss-scale 128.0 ./
5959
$ python -m torch.distributed.launch --nproc_per_node=2 main_amp.py -a resnet50 --b 224 --workers 4 --opt-level O2 ./
6060
```
61-
Options are broken down in detail below.
61+
Options are explained below.
6262

6363
#### `--opt-level O0` (FP32 training) and `O3` (FP16 training)
6464

0 commit comments

Comments
 (0)