numpy
torchvision
timm==0.4.12