Python 3.8.5
tqdm 4.59.0
numpy 1.19.2
pytorch 1.4.0
torchvision 0.5.0
torch_scatter 2.0.4
cudatoolkit 10.1
sh train.sh
For usage of the models, please check out the inference.ipynb file, it has an example.
Python 3.8.5
tqdm 4.59.0
numpy 1.19.2
pytorch 1.4.0
torchvision 0.5.0
torch_scatter 2.0.4
cudatoolkit 10.1
sh train.sh
For usage of the models, please check out the inference.ipynb file, it has an example.