SAG-ViT / requirements.txt
shravvvv's picture
Added model files and updated config.json
039647a
raw
history blame
165 Bytes
numpy==1.26.4
pandas==2.2.3
matplotlib==3.7.5
seaborn==0.12.2
tqdm==4.66.4
scikit-learn==1.2.2
torch==2.4.0
torch-geometric==2.6.1
torchvision==0.19.0
networkx==3.3