An easy-to-use implementation of VICReg in Pytorch.
To rerun experiments or try on your own dataset, first clone the repository and install requirements.txt
.
$ pip install -r requirements.txt
To train model use train.py
.
$ python train.py --path path/to/your/dataset/
- 3 different loss function
- VICReg model and training loop
- Implementation for multiple GPUs
- Compare results with paper