Incorrect Signature of `training_end`, `validation_end`, `test_end` in `Experimet Reporing`
📚 Documentation
Documentation here defines the signature of the functions as follows:
training_end(self, outputs)
test_end(self, outputs)
validation_end(self, outputs)
While in the Experiment Reporting section of docs, the signature of the functions has been misreported as:
training_end(self, batch, batch_idx)
test_end(self, batch, batch_idx)
validation_end(self, batch, batch_idx)