wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 23:08:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Ray SGD] FP16 Hotfix (#10931)

Browse Source
This commit is contained in:
Amog Kamsetty
2020-09-21 13:10:10 -07:00
committed by Barak Michener
parent 7422f64ea9
commit c636f5bd40
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/sgd/torch/training_operator.py
+2 -2
View File
@@ -257,8 +257,8 @@ class TrainingOperator:
if self.use_fp16 and amp:
logger.debug("Setting up Apex.")
self._models, self._optimizers = amp.initialize(
self._models, self._optimizers, **self._apex_args)
self._original_models, self._optimizers = amp.initialize(
self._original_models, self._optimizers, **self._apex_args)
self._amp = amp
if self._wrap_ddp: