Thanks for your work! In the code for image2image_loss, why set vit_embeds.requires_grad = True?
Thanks for your work!
In the code for image2image_loss, why set vit_embeds.requires_grad = True?