args should come after the last positional argument (#1807)

Lightning-AI · May 13, 2020 · e961f7e · e961f7e
1 parent fddd618
commit e961f7e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1439,11 +1439,12 @@ def load_from_metrics(cls, weights_path, tags_csv, map_location=None):
     def load_from_checkpoint(
             cls,
             checkpoint_path: str,
+            *args,
             map_location: Optional[Union[Dict[str, str], str, torch.device, int, Callable]] = None,
             hparams_file: Optional[str] = None,
             tags_csv: Optional[str] = None,  # backward compatible, todo: remove in v0.9.0
             hparam_overrides: Optional[Dict] = None,
-            *args, **kwargs
+            **kwargs
     ) -> 'LightningModule':
         r"""
         Primary way of loading a model from a checkpoint. When Lightning saves a checkpoint