Commit ba429e51 authored by enricoschroeder's avatar enricoschroeder Committed by mcarilli
Browse files

Fix 'local variable 'optimizers_was_list' referenced before assignment' when...

Fix 'local variable 'optimizers_was_list' referenced before assignment' when amp.initialize() is called with optimizers=None (#218)
parent 5b8faa29
...@@ -126,8 +126,8 @@ def _initialize(models, optimizers, properties): ...@@ -126,8 +126,8 @@ def _initialize(models, optimizers, properties):
from apex.parallel import DistributedDataParallel as apex_DDP from apex.parallel import DistributedDataParallel as apex_DDP
from .amp import init as amp_init from .amp import init as amp_init
optimizers_was_list = False
if isinstance(optimizers, torch.optim.Optimizer): if isinstance(optimizers, torch.optim.Optimizer):
optimizers_was_list = False
optimizers = [optimizers] optimizers = [optimizers]
elif optimizers is None: elif optimizers is None:
optimizers = [] optimizers = []
......
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