search

opeltre / revert

Revert
MIT License
2 stars 0 forks source link

Moretwins #23

Closed opeltre closed 2 years ago

opeltre commented 2 years ago

add VICReg class subclassing the general parent class Twins for siamese networks, closing #21

Other losses could be easily implemented, e.g. SimCLR or the like.

Possible generalization: look into student-teacher networks such as BYOL

opeltre commented 2 years ago

N.B: Twins signature has changed: [n, 2, d] -> [n, 2, d'] for consistency with the Module class