set-transformer
set-transformer copied to clipboard
TropComplique
A neural network architecture for prediction on sets
Set Transformer in pytorch
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.
Metadata
21
Stars
3
Forks
Watchers
Owner
TropComplique
Metadata
A neural network architecture for prediction on sets