torch==1.10.0
torchvision==0.7.0
numpy==1.19.2
sklearn==0.23.2
networkx==2.5
