Skip to content

GitLab

Commit a21e633a authored by Michael Carilli's avatar Michael Carilli
Browse files

Merge branch 'master' of https://github.com/NVIDIA/apex

parents 4606df98 cfa5a657
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/amp/opt.py
View file @ a21e633a
...@@ -13,7 +13,7 @@ class OptimWrapper(object): ...@@ -13,7 +13,7 @@ class OptimWrapper(object):
self._num_loss = num_loss self._num_loss = num_loss
self._loss_idx = 0 self._loss_idx = 0
self._skip_next = [False] * num_loss self._skip_next = [False] * num_loss
self._loss_scaler = [LossScaler() for _ in range(num_loss)] self._loss_scaler = [LossScaler('dynamic') for _ in range(num_loss)]
@contextlib.contextmanager @contextlib.contextmanager
def scale_loss(self, loss): def scale_loss(self, loss):
......
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