themrzmaster / git-re-basin-pytorch

Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch
MIT License
72 stars 9 forks source link

change batchnorm of vgg #4

Open themrzmaster opened 2 years ago

themrzmaster commented 2 years ago

https://github.com/themrzmaster/git-re-basin-pytorch/blob/b20023053bf579d7cfd0fcd0a09bec58b1c5faab/models/vgg.py#L33