Skip to content

Commit de1e9d2

Browse files
author
Beat Buesser
committed
Fix line length
Signed-off-by: Beat Buesser <[email protected]>
1 parent 58c0a82 commit de1e9d2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

art/estimators/classification/pytorch.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)