Skip to content

Commit

Permalink
Typo
Browse files Browse the repository at this point in the history
  • Loading branch information
carmocca committed Sep 6, 2021
1 parent ec3dccd commit f078e29
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_lightning/core/lightning.py
Expand Up @@ -632,7 +632,7 @@ def training_step(self, *args, **kwargs) -> STEP_OUTPUT:
- ``None`` - Training will skip to the next batch
Note:
Returning ``None`` is currently not supported for multi-GPU or TPU, or using `DeepSpeed`.
Returning ``None`` is currently not supported for multi-GPU or TPU, or using ``DeepSpeed``.
In this step you'd normally do the forward pass and calculate the loss for a batch.
You can also do fancier things like multiple forward passes or something model specific.
Expand Down

0 comments on commit f078e29

Please sign in to comment.