From 24e55b34d89cff0bc442f7b0b528935e8efc771f Mon Sep 17 00:00:00 2001 From: Peter Yu <2057325+yukw777@users.noreply.github.com> Date: Tue, 12 May 2020 19:21:12 -0400 Subject: [PATCH] args should come after the last positional argument --- pytorch_lightning/core/lightning.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 98966bb988507..c40b6cdb307d8 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -1438,11 +1438,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