Antoni Baum
|
2fb10e6730
|
[SGD] Add support for native Torch AMP in SGD (#16382)
* SGD native AMP initial commit
* SGD native amp second pass
* Update docs
* Update TorchTrainer doc
* Temp fix release test
* Update release/sgd_tests/sgd_gpu/sgd_gpu_app_config.yaml
Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com>
|
2021-06-15 17:48:21 -07:00 |
|