sthalles / PyTorch-BYOL

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

模型的参数需要初始化吗?你的代码里好像没有初始化的步骤? #7

Open LiDaiY opened 3 years ago

hafnerfe commented 3 years ago

@LiDaiY if you initialized the parameters of the resnet with the flag pretrained=True then the model would already have learnt a representation. You might get better results, but the purpose of this repository is to learn a good representation from scratch using BYOL. The other model parameters (classifier, mlp head, ...) are initialized randomly (there is not really any other option).