kakaobrain / torchlars
A LARS implementation in PyTorch
☆345Updated 5 years ago
Alternatives and similar repositories for torchlars:
Users that are interested in torchlars are comparing it to the libraries listed below
- ☆325Updated 2 weeks ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆428Updated 7 months ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆347Updated 2 weeks ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆142Updated 11 months ago
- Understanding Training Dynamics of Deep ReLU Networks☆291Updated 3 months ago
- Unofficial Pytorch code for "FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence" in NeurIPS'20. This repo co…☆190Updated 4 years ago
- A Pytorch-Lightning implementation of self-supervised algorithms☆538Updated 3 years ago
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆423Updated 2 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆203Updated last year
- Estimate/count FLOPS for a given neural network using pytorch☆303Updated 2 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆491Updated 2 years ago
- ☆596Updated 6 months ago
- NeurIPS 2020, Debiased Contrastive Learning☆282Updated 2 years ago
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆414Updated 4 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- Implementation of Barlow Twins paper☆99Updated 2 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆218Updated last month
- Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)☆541Updated 9 months ago
- Experimental ground for optimizing memory of pytorch models☆365Updated 7 years ago
- Distributed, mixed-precision training with PyTorch☆90Updated 4 years ago
- A PyTorch converter for SimCLR checkpoints☆109Updated 4 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆637Updated 2 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆374Updated 4 years ago
- Labels and other data for the paper "Are we done with ImageNet?"☆191Updated 3 years ago
- EsViT: Efficient self-supervised Vision Transformers☆410Updated last year
- Differentiable Data Augmentation Library☆123Updated 2 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆373Updated last year
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆159Updated 3 years ago
- Compare neural networks by their feature similarity☆359Updated last year
- Public repo for Augmented Multiscale Deep InfoMax representation learning☆400Updated 4 years ago