Implementation to calculate loss using Feature Matching.
#1
by
olafjunior9
- opened
I just made an update to my custom implementation to calculate loss using Dino webssl version.
I just did a quick test for 300m but it seen to be working, I might need to improve a little for it, but it should work for all webssl versions I guess.
https://github.com/BurguerJohn/RepAlignLoss/blob/main/RepAlignLoss.py