Fixing Function Signatures (#871)

docs/source/experiment_reporting.rst

@@ -38,23 +38,23 @@ To plot metrics into whatever logger you passed in (tensorboard, comet, neptune,
 
 .. code-block:: python
 
-   def training_end(self, batch, batch_idx):
+   def training_end(self, outputs):
       loss = some_loss()
       ...
 
       logs = {'train_loss': loss}
       results = {'log': logs}
       return results
 
-   def validation_end(self, batch, batch_idx):
+   def validation_end(self, outputs):
       loss = some_loss()
       ...
 
       logs = {'val_loss': loss}
       results = {'log': logs}
       return results
 
-   def test_end(self, batch, batch_idx):
+   def test_end(self, outputs):
       loss = some_loss()
       ...
 
@@ -99,7 +99,7 @@ Here we show the validation loss in the progress bar
 
 .. code-block:: python
 
-   def validation_end(self, batch, batch_idx):
+   def validation_end(self, outputs):
       loss = some_loss()
       ...