torch>=1.4
torchvision
timm>=0.1.28
