CyberZHG / torch-multi-head-attentionLinks
Multi-head attention in PyTorch
☆154Updated 6 years ago
Alternatives and similar repositories for torch-multi-head-attention
Users that are interested in torch-multi-head-attention are comparing it to the libraries listed below
Sorting:
- Implement the paper "Self-Attention with Relative Position Representations"☆139Updated 4 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆221Updated 2 years ago
- A pytorch &keras implementation and demo of Fastformer.☆190Updated 3 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆181Updated 4 years ago
- Implementing SYNTHESIZER: Rethinking Self-Attention in Transformer Models using Pytorch☆70Updated 5 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆271Updated 3 years ago
- ☆84Updated 6 years ago
- PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)☆88Updated 3 years ago
- A quick walk-through of the innards of LSTMs and a naive implementation of the Mogrifier LSTM paper in PyTorch☆78Updated 5 years ago
- Pytorch implementation of R-Transformer. Some parts of the code are adapted from the implementation of TCN and Transformer.☆230Updated 6 years ago
- A collection of awesome things about mixed sample data augmentation☆131Updated 5 years ago
- Loss and accuracy go opposite ways...right?☆95Updated 5 years ago
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆167Updated 4 years ago
- MSc group project: Reproduction of 'Multi-Task Learning using Uncertainty to Weigh Losses for Scene Geometry and Semantics'; A. Kendall, …☆91Updated 6 years ago
- Independent implementation of Supervised Contrastive Loss. Straight to the point and beyond☆85Updated 4 years ago
- A PyTorch implementation of the paper - "Synthesizer: Rethinking Self-Attention in Transformer Models"☆73Updated 2 years ago
- A minimal pytorch package implementing a gradient reversal layer.☆158Updated last year
- Minimal RNN classifier with self-attention in Pytorch☆152Updated 3 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆196Updated 2 years ago
- Reproducing the Linear Multihead Attention introduced in Linformer paper (Linformer: Self-Attention with Linear Complexity)☆75Updated 5 years ago
- ECML 2019: Graph Neural Networks for Multi-Label Classification☆90Updated last year
- PyTorch implementation of "Effective Approaches to Attention-based Neural Machine Translation" using scheduled sampling to improve the pa…☆39Updated 8 years ago
- Custom loss functions to use in (mainly) PyTorch.☆39Updated 5 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆260Updated 4 years ago
- implement Domain-Adversarial Training of Neural Networks☆60Updated 7 years ago
- PyTorch implementation of Pay Attention to MLPs☆40Updated 4 years ago
- Simple implement dilated LSTM, residual LSTM and Attention LSTM (follow the corresponding papers).☆17Updated 5 years ago
- ☆20Updated 6 years ago
- Code release for "LogME: Practical Assessment of Pre-trained Models for Transfer Learning" (ICML 2021) and Ranking and Tuning Pre-trained…☆210Updated 2 years ago
- Code for "Understanding and Improving Layer Normalization"☆46Updated 5 years ago