def validation_epoch_end(self, outputs):
    self.next_token_metrics_epoch_end(outputs, stage='val')
def validation_epoch_end(self, outputs):
    self.next_token_metrics_epoch_end(outputs, stage='val')