chrisby / torchMTLLinks
A lightweight module for Multi-Task Learning in pytorch.
☆168Updated 2 years ago
Alternatives and similar repositories for torchMTL
Users that are interested in torchMTL are comparing it to the libraries listed below
Sorting:
- Metrics for model evaluation in pytorch☆110Updated 4 years ago
- Implementation of Online Label Smoothing in PyTorch☆95Updated 3 years ago
- Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.☆150Updated 6 years ago
- Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."☆132Updated 4 years ago
- ☆148Updated 4 years ago
- Experimenting with different regression losses. Implemented in Pytorch.☆148Updated 6 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 4 years ago
- PyTorch, PyTorch Lightning framework for trying knowledge distillation in image classification problems☆32Updated last year
- Ordinal regression models in PyTorch☆148Updated 3 years ago
- Loss and accuracy go opposite ways...right?☆95Updated 5 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆118Updated 5 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆261Updated 4 years ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆167Updated last year
- Bootstrapping loss function implementation in pytorch☆36Updated 5 years ago
- A deep learning framework for building multimodal multi-task learning systems.☆112Updated 2 years ago
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated last year
- Simple pytorch implementation of focal loss☆86Updated 2 years ago
- Loss function which directly targets ROC-AUC☆250Updated 2 years ago
- PyTorch samplers that output roughly balanced batches with support for multilabel datasets☆57Updated last year
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆169Updated 3 years ago
- Minimalistic TensorFlow2+ deep metric/similarity learning library with loss functions, miners, and utils as embedding projector.☆38Updated 2 years ago
- A list of multi-task learning papers and projects.☆383Updated 3 years ago
- An education step by step implementation of SimCLR that accompanies the blogpost☆31Updated 3 years ago
- Unofficial implementation of Switching from Adam to SGD optimization in PyTorch.☆68Updated 3 years ago
- Simple stochastic weight averaging callback for Keras☆63Updated 4 years ago
- ☆34Updated 5 years ago
- This Module implements Spatial Pyramid Pooling (SPP) and Temporal Pyramid Pooling (TPP) as described in different papers.☆58Updated 6 years ago
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆239Updated last year
- NeuralPy: A Keras like deep learning library works on top of PyTorch☆79Updated last year
- Implementation of self-supervised image-level contrastive pretraining methods using Keras.☆70Updated 4 years ago