Add the two loss functions instead of multiplying them

lucidrains / pixel-level-contrastive-learning

Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch

MIT License

252 stars 29 forks source link

Closed cjrd closed 3 years ago

cjrd commented 3 years ago

I take it these two loss functions are from Eq. (6) in the paper, where they are added together instead of multiplied.

lucidrains commented 3 years ago

omg yes, lol