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 020adeb commit a6179aeCopy full SHA for a6179ae
pytorch_lightning/accelerators/accelerator.py
@@ -195,9 +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`): Passed in if
199
- :paramref:`~pytorch_lightning.core.lightning.LightningModule.truncated_bptt_steps` > 0
200
-
+ - hiddens(:class:`~torch.Tensor`):
+ Passed in if :paramref:`~pytorch_lightning.core.lightning.LightningModule.truncated_bptt_steps` > 0
201
"""
202
args[0] = self.to_device(args[0])
203
0 commit comments