AdeelH / pytorch-multi-class-focal-lossLinks
An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
☆238Updated 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
Sorting:
- Unofficial PyTorch implementation of "Meta Pseudo Labels"☆389Updated last year
- Learning Rate Warmup in PyTorch☆413Updated 4 months ago
- ☆466Updated 2 years ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆337Updated 3 years ago
- Self-supervised vIsion Transformer (SiT)☆338Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆437Updated last year
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆169Updated 3 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆221Updated 2 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆800Updated last year
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆991Updated last year
- Official Implementation of Early-Learning Regularization Prevents Memorization of Noisy Labels☆296Updated 2 years ago
- Official Pytorch Implementation of: "Asymmetric Loss For Multi-Label Classification"(ICCV, 2021) paper☆774Updated 2 years ago
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆309Updated 3 years ago
- Reproduce Results for ICCV2019 "Symmetric Cross Entropy for Robust Learning with Noisy Labels" https://arxiv.org/abs/1908.06112☆189Updated 4 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆636Updated 2 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆149Updated 4 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆263Updated 4 years ago
- A PyTorch Implementation of Focal Loss.☆988Updated 6 years ago
- A Pytorch-Lightning implementation of self-supervised algorithms☆546Updated 3 years ago
- Implementation of Visual Transformer for Small-size Datasets☆126Updated 3 years ago
- (ICCV 2021 Oral) CoaT: Co-Scale Conv-Attentional Image Transformers☆233Updated 3 years ago
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆386Updated 4 years ago
- EsViT: Efficient self-supervised Vision Transformers☆412Updated 2 years ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆765Updated 4 years ago
- ☆254Updated 4 years ago
- Code for the Convolutional Vision Transformer (ConViT)☆469Updated 3 years ago
- Gradient Reversal Layer for Domain Adaptation☆127Updated 2 years ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆349Updated last year
- ☆72Updated 4 years ago
- Simple pytorch implementation of focal loss☆86Updated 2 years ago