torch
tqdm
torch-geometric
transformers
scipy
numpy
cv2