diff --git a/src/lightning/pytorch/callbacks/pruning.py b/src/lightning/pytorch/callbacks/pruning.py index f0e1bcbe49f99..9e646edc3a81d 100644 --- a/src/lightning/pytorch/callbacks/pruning.py +++ b/src/lightning/pytorch/callbacks/pruning.py @@ -79,7 +79,7 @@ def __init__( verbose: int = 0, prune_on_train_epoch_end: bool = True, ) -> None: - """Model pruning Callback, using PyTorch's prune utilities. This callback is responsible of pruning networks + """Model pruning Callback, using PyTorch's prune utilities. This callback is responsible for pruning networks parameters during training. To learn more about pruning with PyTorch, please take a look at @@ -444,7 +444,7 @@ def on_save_checkpoint(self, trainer: "pl.Trainer", pl_module: LightningModule, def sanitize_parameters_to_prune( pl_module: LightningModule, parameters_to_prune: _PARAM_LIST = (), parameter_names: Sequence[str] = () ) -> _PARAM_LIST: - """This function is responsible of sanitizing ``parameters_to_prune`` and ``parameter_names``. If + """This function is responsible for sanitizing ``parameters_to_prune`` and ``parameter_names``. If ``parameters_to_prune is None``, it will be generated with all parameters of the model. Raises: