Code for training target model

[thomascong121]

Thanks for the great work. I would like to try your idea on a custom dataset, but it reads to me that I need to first train target models in the self-supervised manner. Do you have the code for this part?