Skip to content

Commit d1feee8

Browse files
gsgandenGreg Gandenberger
authored andcommitted
Fix argument order
1 parent b8698d8 commit d1feee8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/source/trainer/using_the_trainer.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ minimally implementing the following methods:
3333

3434
- ``def forward(batch) -> outputs`` : computes the forward pass based
3535
on the ``batch`` returned from the dataloader.
36-
- ``def loss(batch, outputs)``: returns the loss based on the
36+
- ``def loss(outputs, batch)``: returns the loss based on the
3737
``outputs`` from the forward pass and the dataloader.
3838

3939
For more information, see the :doc:`ComposerModel</composer_model>` guide.

0 commit comments

Comments
 (0)