search

PatrickHua / SimSiam

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

Update simsiam.py #41

Open hhhhnwl opened 3 years ago

hhhhnwl commented 3 years ago

if hidden_dim and out_dim is diffierent in projection_MLP's layer3, BatchNorm1d's param should be out_dim