HalbertCH / IEContraAST

This is the official PyTorch implementation of our paper: "Artistic Style Transfer with Internal-external Learning and Contrastive Learning".
MIT License
78 stars 7 forks source link

Is there a mix-up in the calculation of style_contrastive_loss and content_contrastive_loss? #7

Open wanghaohw opened 7 months ago

wanghaohw commented 7 months ago

I find the code calculate the style_contrastive_loss and content_contrastive_loss have something that bothers me at line 325 and 360 in Net.py, Are they upside down? 屏幕截图 2023-11-21 104056 屏幕截图 2023-11-21 104124

llzygg commented 7 months ago

请问这里的up和down的含义,而且对比损失似乎形式与论文不一样,麻烦帮忙解答一下,谢谢