args should come after the last positional argument (#1807)
This commit is contained in:
parent
fddd618915
commit
e961f7e344
|
@ -1439,11 +1439,12 @@ class LightningModule(ABC, DeviceDtypeModuleMixin, GradInformation, ModelIO, Mod
|
|||
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
|
||||
|
|
Loading…
Reference in New Issue