AntixK / PyTorch-Model-Compare
Compare neural networks by their feature similarity
☆356Updated 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☆299Updated 4 years ago
- PyTorch implementation for Vision Transformer[Dosovitskiy, A.(ICLR'21)] modified to obtain over 90% accuracy FROM SCRATCH on CIFAR-10 wit…☆184Updated last year
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆343Updated 7 months ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆207Updated last year
- Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch☆323Updated last year
- Code for coreset selection methods☆230Updated 2 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- An Numpy and PyTorch Implementation of CKA-similarity with CUDA support☆90Updated 3 years ago
- Awesome coreset/core-set/subset/sample selection works.☆173Updated 9 months ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆141Updated 10 months ago
- Transformers trained on Tiny ImageNet☆54Updated 2 years ago
- Probing the representations of Vision Transformers.☆321Updated 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)☆143Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆567Updated 4 months ago
- This is a curated list for Information Bottleneck Principle, in memory of Professor Naftali Tishby.☆351Updated 10 months ago
- Official code for our CVPR '22 paper "Dataset Distillation by Matching Training Trajectories"☆419Updated 8 months ago
- A curated list of papers in Test-time Adaptation, Test-time Training and Source-free Domain Adaptation☆488Updated 9 months ago
- ImageNet-R(endition) and DeepAugment (ICCV 2021)☆263Updated 3 years ago
- This repository contains the implementation for the paper "EMP-SSL: Towards Self-Supervised Learning in One Training Epoch."☆225Updated last year
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆203Updated 11 months ago
- ☆592Updated 5 months ago
- Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time☆454Updated 8 months ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆129Updated 4 months ago
- A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…☆621Updated 2 years ago
- A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...☆317Updated 2 weeks ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆728Updated 11 months ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,071Updated 2 years ago
- [BMVC 2022] Official repository for "How to Train Vision Transformer on Small-scale Datasets?"☆149Updated last year
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆149Updated 2 years ago