attrdict==2.0.1
torch>=1.4.0
torch-geometric>=1.4.2
torch-scatter>=2.0.4
torch-sparse>=0.6.0
torchvision>=0.5.0
sklearn

