add vicreg, add cifar-10, refactor, small fixes - Githubissues

imbue-ai / self_supervised

A Pytorch-Lightning implementation of self-supervised algorithms

MIT License

536 stars 52 forks source link

add vicreg, add cifar-10, refactor, small fixes #13

Closed ekitanidis closed 2 years ago

ekitanidis commented 2 years ago

Add VICReg and refactor methods and model params
Add CIFAR-10 and refactor datasets and data transforms
Make LARS eta into hyperparam
Add more logging
Fix bug from pytorch-lightning versions >= 1.3.0
Update README