diff --git a/pytorch_lightning/utilities/cli.py b/pytorch_lightning/utilities/cli.py index 35677d35577f8..459217786d84e 100644 --- a/pytorch_lightning/utilities/cli.py +++ b/pytorch_lightning/utilities/cli.py @@ -194,7 +194,7 @@ def __init__( parser_kwargs: Dict[str, Any] = None, subclass_mode_model: bool = False, subclass_mode_data: bool = False, - skip_fit: bool = False, + instantiate_only: bool = False, ) -> None: """ Receives as input pytorch-lightning classes, which are instantiated @@ -238,7 +238,7 @@ def __init__( subclass_mode_data: Whether datamodule can be any `subclass `_ of the given class. - skip_fit: If ``True``, the CLI won't automatically call ``Trainer.fit(...)``. + instantiate_only: If ``True``, the CLI won't automatically call ``Trainer.fit(...)``. """ assert issubclass(trainer_class, Trainer) assert issubclass(model_class, LightningModule) @@ -269,7 +269,7 @@ def __init__( self.add_configure_optimizers_method_to_model() self.prepare_fit_kwargs() - if skip_fit: + if instantiate_only: return self.before_fit() self.fit()