Issue with [ def validation_epoch_end(self, outputs): ]

[Wadha-Almattar]

Hi there, I'm trying to re-produce the work. Unfortunately, I'm getting this issue each time I run the code:

TypeError: SelfSupervisedMethod.on_validation_epoch_end() missing 1 required positional argument: 'outputs'

I've seek the issue but could't find a proper solution!
It seems that on_validation_epoch_end() mentioned in the error ---> with [on] , receives only one argument
but as we're not using it in the code, still it's called somewhere from the lighting package ?

Any suggestions would be appreciated,

Thanks,

[Tieck-IT]

I encountered the same issue. This is my solution using a small trick.

def validation_step(self, batch, batch_idx):

        self.outputs = {"emb": emb, "labels": class_labels}

        
def on_validation_epoch_end(self):

        outputs = self.outputs