Skip to content

Commit

Permalink
Update art/estimators/certification/derandomized_smoothing/pytorch.py
Browse files Browse the repository at this point in the history
  • Loading branch information
beat-buesser authored Dec 19, 2023
1 parent 804af32 commit 94cf59f
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit 94cf59f

Please sign in to comment.