PyTorch Lightningのepoch_endというフックがなくなった
Exampleもある
validation_epoch_endを例に
初期化時にself.validation_step_outputs = []を持たせておく
validation_stepでappendする
on_validation_epoch_endでself.validation_step_outputsのロスを平均する。その後clear
torch.stack(self.validation_step_outputs).mean()
V2で消えた