torch==2.3.1
matplotlib==3.1.3
argparse==1.4.0
numpy==1.24.4
scipy==1.10.1
pyyaml==5.3.1
torch_geometric==2.6.1
torch_sparse==0.6.18+pt23cu121
sklearn==1.3.2
pandas==1.0.1
torchmetrics==1.4.2
torchvision==0.18.1+cu118
