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 b8698d8 commit d1feee8Copy full SHA for d1feee8
docs/source/trainer/using_the_trainer.rst
@@ -33,7 +33,7 @@ minimally implementing the following methods:
33
34
- ``def forward(batch) -> outputs`` : computes the forward pass based
35
on the ``batch`` returned from the dataloader.
36
-- ``def loss(batch, outputs)``: returns the loss based on the
+- ``def loss(outputs, batch)``: returns the loss based on the
37
``outputs`` from the forward pass and the dataloader.
38
39
For more information, see the :doc:`ComposerModel</composer_model>` guide.
0 commit comments