MrChenFeng / SSR_BMVC2022

SSR: An Efficient and Robust Framework for Learning with Unknown Label Noise (BMVC2022)
https://bmvc2022.mpi-inf.mpg.de/372/
MIT License
29 stars 7 forks source link

Q for D #7

Closed knif-knif closed 5 months ago

knif-knif commented 5 months ago

utils/funcs.py line 30

return - F.cosine_similarity(p, z.detach(), dim=-1).mean()

It's negative, Should it be

return 1 - F.cosine_similarity(p, z.detach(), dim=-1).mean()
MrChenFeng commented 5 months ago

hi, the constant is omitted as it won't affect the gradient.