chrisby / torchMTLLinks
A lightweight module for Multi-Task Learning in pytorch.
☆164Updated last year
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☆113Updated 4 years ago
- Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."☆134Updated 3 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- Experimenting with different regression losses. Implemented in Pytorch.☆147Updated 6 years ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆161Updated last year
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 4 years ago
- Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.☆149Updated 6 years ago
- PyTorch samplers that output roughly balanced batches with support for multilabel datasets☆57Updated last year
- ☆147Updated 3 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 3 years ago
- A deep learning framework for building multimodal multi-task learning systems.☆110Updated last year
- Loss and accuracy go opposite ways...right?☆94Updated 5 years ago
- Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.☆238Updated 3 years ago
- Reproducing the Linear Multihead Attention introduced in Linformer paper (Linformer: Self-Attention with Linear Complexity)☆76Updated 5 years ago
- Ordinal regression models in PyTorch☆147Updated 3 years ago
- Loss function which directly targets ROC-AUC☆247Updated 2 years ago
- This Module implements Spatial Pyramid Pooling (SPP) and Temporal Pyramid Pooling (TPP) as described in different papers.☆56Updated 6 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆115Updated 4 years ago
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆236Updated last year
- Pseudo Labeling for Neural Networks and Logistic Regression/SVMs ( Based on "Pseudo-Label : The Simple and Efficient Semi-Supervised Lear…☆73Updated 5 years ago
- Code for Multi-Head Attention: Collaborate Instead of Concatenate☆152Updated 2 years ago
- Bootstrapping loss function implementation in pytorch☆36Updated 4 years ago
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆198Updated 2 years ago
- A list of multi-task learning papers and projects.☆375Updated 3 years ago
- Simple pytorch implementation of focal loss☆86Updated 2 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆141Updated 4 years ago
- Unofficial implementation of Switching from Adam to SGD optimization in PyTorch.☆66Updated 2 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆178Updated 3 years ago
- PyTorch, PyTorch Lightning framework for trying knowledge distillation in image classification problems☆32Updated 11 months ago
- Official TensorFlow code for the paper "Efficient-CapsNet: Capsule Network with Self-Attention Routing".☆273Updated 3 years ago