narame7/set-transformer

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

1. pytorch 1.3
2. numpy 1.17, matplotlib 3.1, tqdm 4.32
3. (optional) NVIDIA/apex

Credit

This code is partially based on jadore801120/attention-is-all-you-need-pytorch.