One can run training on the main script as follows:
python hypersets.py
TODO
This environment is built using Python 3.8.5 on Ubuntu 20.04 using Anaconda\
Pytorch installation:
conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch
I suggest to also install Pytorch Geometric using Anaconda Link to guide here
conda install pyg -c pyg -c conda-forge