We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a6179ae commit 418cb7eCopy full SHA for 418cb7e
pytorch_lightning/accelerators/accelerator.py
@@ -195,8 +195,8 @@ def training_step(
195
The output of your :class:`~torch.utils.data.DataLoader`. A tensor, tuple or list.
196
- batch_idx (int): Integer displaying index of this batch
197
- optimizer_idx (int): When using multiple optimizers, this argument will also be present.
198
- - hiddens(:class:`~torch.Tensor`):
199
- Passed in if :paramref:`~pytorch_lightning.core.lightning.LightningModule.truncated_bptt_steps` > 0
+ - hiddens(:class:`~torch.Tensor`): Passed in if
+ :paramref:`~pytorch_lightning.core.lightning.LightningModule.truncated_bptt_steps` > 0
200
"""
201
args[0] = self.to_device(args[0])
202
0 commit comments