torch>=1.10
numpy
tqdm
pyyaml
scipy
matplotlib
pillow
