issues
search
facebookresearch
/
vicreg
VICReg official code base
MIT License
525
stars
88
forks
source link
Coefficients of loss
#26
Open
ChicyChen
opened
1 year ago
ChicyChen
commented
1 year ago
Hi, in the code the std loss is averaged by 2 (not the same as the paper) while the cov loss is not (same as the paper). Then, is the coefficient of mse vs. std vs. cov loss still 25.0 & 25.0 & 1.0 (same as the paper), but not 25.0 & 50.0 & 1?
https://github.com/facebookresearch/vicreg/blob/4e12602fd495af83efd1631fbe82523e6db092e0/main_vicreg.py#L211
Thanks a lot!