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