hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-08 19:41:38 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions
ray/release/sgd_tests
History
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
..
sgd_gpu [SGD] Add support for native Torch AMP in SGD (#16382) 2021-06-15 17:48:21 -07:00
sgd_tests.yaml [SGD] Add support for native Torch AMP in SGD (#16382) 2021-06-15 17:48:21 -07:00
wait_cluster.py [SGD] Add support for native Torch AMP in SGD (#16382) 2021-06-15 17:48:21 -07:00