torch>=1.6.0
pyyaml
pandas
einops
scipy
matplotlib
dominate
visdom
