torch>=1.7
torchvision
pyyaml
timm
grad-cam
scikit-learn
matplotlib
