Reza Alipour

Results 2 comments of Reza 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):...