### Requirements
torch 1.5 
pip install local_attention axial_positional_embedding product_key_memory

### Dataset
The pre-processed five datasets are in the "data" folder

### Training and Testing
python main.py --dataName='SUBJ'  --modelName='Clusterformer' --n_centroid=5 --commitment1 1 --commitment2 1


### Time test
python shell_times.py

