Access dataset directly in validation_epoch_end
🚀 Feature

Motivation / Pitch

Not sure if duplicated but:
I would propose to allow users to add additional arguments to be passed into functions such as validation_epoch_end().
One of the examples being: during validation, we might need to fetch some additional information from the dataset (e.g., len(dataset)), and currently it is a bit hard to achieve without passing the dataset into the model during init.