AntixK / PyTorch-Model-Compare
Compare neural networks by their feature similarity
☆349Updated last year
Alternatives and similar repositories for PyTorch-Model-Compare:
Users that are interested in PyTorch-Model-Compare are comparing it to the libraries listed below
- Reproduce CKA: Similarity of Neural Network Representations Revisited☆297Updated 4 years ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆339Updated 5 months ago
- PyTorch implementation for Vision Transformer[Dosovitskiy, A.(ICLR'21)] modified to obtain over 90% accuracy FROM SCRATCH on CIFAR-10 wit…☆177Updated 11 months ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch☆316Updated last year
- Train ImageNet *fast* in 500 lines of code with FFCV☆136Updated 8 months ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆197Updated 9 months ago
- An Numpy and PyTorch Implementation of CKA-similarity with CUDA support☆88Updated 3 years ago
- ImageNet-R(endition) and DeepAugment (ICCV 2021)☆259Updated 3 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆203Updated last year
- ☆575Updated 3 months ago
- Pytorch implementation of the paper Exploring Simple Siamese Representation Learning.☆72Updated 4 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆216Updated 3 years ago
- [NeurIPS 2021] Official codes for "Efficient Training of Visual Transformers with Small Datasets".☆140Updated 3 weeks ago
- ☆227Updated 3 years ago
- ☆314Updated 10 months ago
- [BMVC 2022] Official repository for "How to Train Vision Transformer on Small-scale Datasets?"☆145Updated last year
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆808Updated 2 years ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆139Updated 2 years ago
- Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time☆437Updated 6 months ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆267Updated 2 years ago
- Probing the representations of Vision Transformers.☆319Updated 2 years ago
- Transformers trained on Tiny ImageNet☆50Updated 2 years ago
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆483Updated 3 years ago
- ☆104Updated last year
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated last year
- A LARS implementation in PyTorch☆338Updated 4 years ago
- Official code for our CVPR '22 paper "Dataset Distillation by Matching Training Trajectories"☆409Updated 6 months ago
- Official Implementation of Early-Learning Regularization Prevents Memorization of Noisy Labels☆293Updated last year
- Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch☆73Updated last year