search

sthalles / PyTorch-BYOL

PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning
480 stars 72 forks source link

training on CIFAR-10 #12

Open qiaoviola opened 2 years ago

qiaoviola commented 2 years ago

Hello, Thank you very much for the program you shared, but I want to test it on the CIFAR-10 data set. How should the relevant parameters be set?

Issac-Sun commented 2 days ago

Hello, Thank you very much for the program you shared, but I want to test it on the CIFAR-10 data set. How should the relevant parameters be set? I completed a simple implementation on CIFAR10 and the cosine similarity loss in the original article, you can follow https://github.com/Issac-Sun/BYOL_pytorch