timm numpy matplotlib torch==1.13.1 torchvision==0.14.1