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

a21e633a · Michael Carilli · 4606df98 · cfa5a657 · a21e633a
Commit a21e633a authored Mar 07, 2019 by Michael Carilli
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

apex/amp/opt.py apex/amp/opt.py +1 -1

No files found.
--- a/apex/amp/opt.py
+++ b/apex/amp/opt.py
@@ -13,7 +13,7 @@ class OptimWrapper(object):
        self._num_loss = num_loss
        self._loss_idx = 0
        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
    def scale_loss(self, loss):