pytorch
Set Transformer in This an implementation of Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks.
How to use this code
Just run python train.py
.
Requirements
- pytorch 1.3
- numpy 1.17, matplotlib 3.1, tqdm 4.32
- (optional) NVIDIA/apex
Credit
This code is partially based on jadore801120/attention-is-all-you-need-pytorch.