File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
art/estimators/classification Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -374,11 +374,11 @@ def fit( # pylint: disable=W0221
374374 :param batch_size: Size of batches.
375375 :param nb_epochs: Number of epochs to use for training.
376376 :param training_mode: `True` for model set to training mode and `'False` for model set to evaluation mode.
377- :param drop_last: Set to ``True`` to drop the last incomplete batch, if the dataset size is not divisible by the batch
378- size. If ``False`` and the size of dataset is not divisible by the batch size, then the last batch
379- will be smaller. (default: ``False``)
377+ :param drop_last: Set to ``True`` to drop the last incomplete batch, if the dataset size is not divisible by
378+ the batch size. If ``False`` and the size of dataset is not divisible by the batch size, then
379+ the last batch will be smaller. (default: ``False``)
380380 :param kwargs: Dictionary of framework-specific arguments. This parameter is not currently supported for PyTorch
381- and providing it takes no effect.
381+ and providing it takes no effect.
382382 """
383383 import torch # lgtm [py/repeated-import]
384384
You can’t perform that action at this time.
0 commit comments