vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
MIT License
1.42k stars 185 forks source link

Barlow Twins ImageNet-1k Checkpoint #330

Closed ThomasYerxa closed 1 year ago

ThomasYerxa commented 1 year ago

It seems like there is no actual checkpoint available to download for Barlow Twins pretrained for 100 epochs on ImageNet (the drive folder only contains the args.json folder).

Is it possible to add the actual weights?

vturrisi commented 1 year ago

Hey. That's very strange because we had the checkpoint there for sure. Anyway, I'll try to find it and add it as soon as possible.

vturrisi commented 1 year ago

@ThomasYerxa sorry about the mega-long delay. Had a lot of issues with vpn/finding the checkpoint. Just re-uploaded it to the google drive folder (you will also be able to download it with the zoo/imagenet.sh script.