themrzmaster / git-re-basin-pytorch

Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch

change batchnorm of vgg

themrzmaster opened this issue 2 years ago · comments

Lucca Zenóbio commented 2 years ago

git-re-basin-pytorch/models/vgg.py

Line 33 in b200230

nn.BatchNorm2d(x),