Skip to content

GitLab

Unverified Commit a21a945c authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

Update fp16_optimizer.py

parent 88effd5d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/fp16_utils/fp16_optimizer.py
View file @ a21a945c
...@@ -184,7 +184,7 @@ class FP16_Optimizer(object): ...@@ -184,7 +184,7 @@ class FP16_Optimizer(object):
self.fp32_from_fp16_groups.append(fp32_from_fp16_params_this_group) self.fp32_from_fp16_groups.append(fp32_from_fp16_params_this_group)
self.fp32_from_fp32_groups.append(fp32_params_this_group) self.fp32_from_fp32_groups.append(fp32_params_this_group)
self.optimizer = optimizer self.optimizer = init_optimizer
if dynamic_loss_scale: if dynamic_loss_scale:
self.dynamic_loss_scale = True self.dynamic_loss_scale = True
......
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