Checkpoint fix

3e2883dd · Michael Carilli · 72bce160 · 3e2883dd
Commit 3e2883dd authored May 07, 2019 by Michael Carilli
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

apex/optimizers/fused_sgd.py apex/optimizers/fused_sgd.py +1 -1

No files found.
--- a/apex/optimizers/fused_sgd.py
+++ b/apex/optimizers/fused_sgd.py
@@ -76,7 +76,7 @@ class FusedSGD(Optimizer):
            raise RuntimeError('apex.optimizers.FusedSGD requires cuda extensions')

    def __setstate__(self, state):
-        super(SGD, self).__setstate__(state)
+        super(FusedSGD, self).__setstate__(state)
        for group in self.param_groups:
            group.setdefault('nesterov', False)