Skip to content

Commit

Permalink
Fix order of arguments to loss (#3754)
Browse files Browse the repository at this point in the history
Co-authored-by: Daniel King <[email protected]>
  • Loading branch information
gsganden and dakinggg authored Jan 31, 2025
1 parent 4d71a21 commit 1c20ee4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/trainer/using_the_trainer.rst
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ minimally implementing the following methods:

- ``def forward(batch) -> outputs`` : computes the forward pass based
on the ``batch`` returned from the dataloader.
- ``def loss(batch, outputs)``: returns the loss based on the
- ``def loss(outputs, batch)``: returns the loss based on the
``outputs`` from the forward pass and the dataloader.

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

0 comments on commit 1c20ee4

Please sign in to comment.