diff --git a/art/estimators/certification/derandomized_smoothing/pytorch.py b/art/estimators/certification/derandomized_smoothing/pytorch.py index beb67ed171..52b38d604d 100644 --- a/art/estimators/certification/derandomized_smoothing/pytorch.py +++ b/art/estimators/certification/derandomized_smoothing/pytorch.py @@ -457,7 +457,7 @@ def fit( # pylint: disable=W0221 the batch size. If ``False`` and the size of dataset is not divisible by the batch size, then the last batch will be smaller. (default: ``False``) :param scheduler: Learning rate scheduler to run at the start of every epoch. - :param verbose: if to display training progress bars + :param verbose: Display training progress bar. :param update_batchnorm: ViT specific argument. If to run the training data through the model to update any batch norm statistics prior to training. Useful on small datasets when using pre-trained ViTs.