TypeError: on_train_epoch_end() missing 1 required positional argument: 'outputs'

[m-ali-awan]

As soon as the first epoch completes training, I am getting this error. I think it is due to some changes in the updated library, as I am getting the same error with one more implementation on vae.

Kindly help me on this.

Thanks, Regards, M.Ali

[mathemusician]

@m-ali-awan You have to use on_train_epoch_end(self, *arg, **kwargs):

[Abolfazl-Alipour]

just delete the outputs from the required variables.

[RachelWang122]

hi，Have you solved this problem, can you tell me the specific operation, thank you very much

[Abolfazl-Alipour]

so I got this error when calling "on_train_epoch_end(self, trainer, pl_module, outputs):"

you need to delete the 'outputs' as an input and just call the function this way: on_train_epoch_end(self, trainer, pl_module): seems like an update issue with newer versions on pytorch lightining.

[CiaoHe]

this is funny... just fail after a long time