Unverified Commit 5b71d369 authored by Ayla Khan's avatar Ayla Khan Committed by GitHub
Browse files

Fix opt_level command line arg in instructions. (#713)

Actual flag is --opt_level and copy pasting the example results in an unrecognized arguments error.
parent 3f94528e
......@@ -29,13 +29,13 @@ With the new Amp API **you never need to explicitly convert your model, or the i
"Pure FP32" training:
```
$ python main_amp.py --opt-level O0
$ python main_amp.py --opt_level O0
```
Recommended mixed precision training:
```
$ python main_amp.py --opt-level O1
$ python main_amp.py --opt_level O1
```
Have a look at the original [DCGAN example](https://github.com/pytorch/examples/tree/master/dcgan) for more information about the used arguments.
To enable mixed precision training, we introduce the `--opt-level` argument.
To enable mixed precision training, we introduce the `--opt_level` argument.
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