search

themrzmaster / git-re-basin-pytorch

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

add resnet20 and use conv + norm in shortcut #1

Closed themrzmaster closed 2 years ago

themrzmaster commented 2 years ago

change for a custom resnet implementation with this kind of shortcut https://github.com/samuela/git-re-basin/blob/main/src/resnet20.py#L34