search

PatrickHua / SimSiam

A pytorch implementation for paper 'Exploring Simple Siamese Representation Learning'
MIT License
814 stars 135 forks source link

Some small mistakes #42

Open Levigty opened 3 years ago

Levigty commented 3 years ago

Thanks for your work. I thank the "_nn.BatchNorm1d(hiddendim)" in line 42 in the models/simsiam.py should be changed to "_nn.BatchNorm1d(outdim)".