How to add input variables in the forward process?

Hi, I encounter a problem when modifying the ssl method of BYOL. Assume there comes a batch of 256 images in the forward function, I want to also include the corresponding ground truth labels as the input in the forward function? How can I realize this? Seems like the data preprocessing pipeline has assumed to be fixed?
