How to add input variables in the forward process?

open-mmlab / mmselfsup

OpenMMLab Self-Supervised Learning Toolbox and Benchmark

https://mmselfsup.readthedocs.io/en/latest/

Apache License 2.0

3.14k stars 429 forks source link

How to add input variables in the forward process? #765

Open kasteric opened 1 year ago

kasteric commented 1 year ago

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?