Skip to content

Commit a6179ae

Browse files
committed
Update accelerator.py
1 parent 020adeb commit a6179ae

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pytorch_lightning/accelerators/accelerator.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -195,9 +195,8 @@ def training_step(
195195
The output of your :class:`~torch.utils.data.DataLoader`. A tensor, tuple or list.
196196
- batch_idx (int): Integer displaying index of this batch
197197
- optimizer_idx (int): When using multiple optimizers, this argument will also be present.
198-
- hiddens(:class:`~torch.Tensor`): Passed in if
199-
:paramref:`~pytorch_lightning.core.lightning.LightningModule.truncated_bptt_steps` > 0
200-
198+
- hiddens(:class:`~torch.Tensor`):
199+
Passed in if :paramref:`~pytorch_lightning.core.lightning.LightningModule.truncated_bptt_steps` > 0
201200
"""
202201
args[0] = self.to_device(args[0])
203202

0 commit comments

Comments
 (0)