kakaobrain / torchlars
A LARS implementation in PyTorch
☆338Updated 4 years ago
Alternatives and similar repositories for torchlars:
Users that are interested in torchlars are comparing it to the libraries listed below
- Unofficial Pytorch code for "FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence" in NeurIPS'20. This repo co…☆190Updated 3 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆418Updated 3 months ago
- ☆446Updated last year
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆413Updated 4 years ago
- Understanding Training Dynamics of Deep ReLU Networks☆284Updated 2 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆480Updated 2 years ago
- Differentiable Data Augmentation Library☆121Updated 2 years ago
- Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)☆538Updated 5 months ago
- Experimental ground for optimizing memory of pytorch models☆361Updated 6 years ago
- ☆314Updated 9 months ago
- Implementation of https://arxiv.org/abs/1904.00962☆372Updated 4 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆631Updated last year
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆339Updated 5 months ago
- Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)☆159Updated 3 years ago
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆136Updated 8 months ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆981Updated 5 months ago
- Estimate/count FLOPS for a given neural network using pytorch☆304Updated 2 years ago
- A Pytorch-Lightning implementation of self-supervised algorithms☆536Updated 2 years ago
- ☆575Updated 2 months ago
- Learning Rate Warmup in PyTorch☆399Updated this week
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆197Updated 8 months ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆579Updated 8 months ago
- NeurIPS 2020, Debiased Contrastive Learning☆282Updated last year
- A PyTorch converter for SimCLR checkpoints☆107Updated 3 years ago
- Scaling and Benchmarking Self-Supervised Visual Representation Learning☆585Updated 3 years ago
- A new test set for ImageNet☆243Updated last year
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆483Updated 3 years ago