AdeelH / pytorch-multi-class-focal-loss
An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
☆233Updated last year
Alternatives and similar repositories for pytorch-multi-class-focal-loss:
Users that are interested in pytorch-multi-class-focal-loss are comparing it to the libraries listed below
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆333Updated 3 years ago
- ☆450Updated last year
- Learning Rate Warmup in PyTorch☆404Updated 3 weeks ago
- Unofficial PyTorch implementation of "Meta Pseudo Labels"☆386Updated last year
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆792Updated last year
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆426Updated 5 months ago
- Experiments with supervised contrastive learning methods with different loss functions☆219Updated 2 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆259Updated 4 years ago
- A PyTorch Implementation of Focal Loss.☆980Updated 5 years ago
- A LARS implementation in PyTorch☆343Updated 5 years ago
- [ ECCV 2020 Spotlight ] Pytorch implementation for "Distribution-Balanced Loss for Multi-Label Classification in Long-Tailed Datasets"☆366Updated 2 years ago
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆304Updated 3 years ago
- Self-supervised vIsion Transformer (SiT)☆327Updated 2 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆133Updated 4 years ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆345Updated last year
- Code for the Convolutional Vision Transformer (ConViT)☆467Updated 3 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆418Updated 10 months ago
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆168Updated 2 years ago
- EsViT: Efficient self-supervised Vision Transformers☆410Updated last year
- Unofficial PyTorch Reimplementation of RandAugment.☆633Updated 2 years ago
- ☆245Updated 3 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆167Updated 2 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆490Updated 2 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆988Updated 5 months ago
- Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)☆457Updated 2 years ago
- This is an official implementation for "Self-Supervised Learning with Swin Transformers".☆647Updated 3 years ago
- Official Implementation of Early-Learning Regularization Prevents Memorization of Noisy Labels☆295Updated last year
- ☆237Updated 3 years ago
- A Pytorch-Lightning implementation of self-supervised algorithms☆537Updated 2 years ago