added state saving

Lightning-AI · Apr 23, 2020 · c60a885 · c60a885
1 parent fa87d1d
commit c60a885
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/training_io.py b/pytorch_lightning/trainer/training_io.py
@@ -322,7 +322,7 @@ def dump_checkpoint(self):
 
         # restore native amp scaling
         if self.use_amp and self.use_native_amp and 'native_amp_scaling_state' in checkpoint:
-            checkpoint['native_amp_scaling_state'] = self.scaler.state_dict
+            checkpoint['native_amp_scaling_state'] = self.scaler.state_dict()
 
         if hasattr(model, "hparams"):
             is_namespace = isinstance(model.hparams, Namespace)