torch == 2.0.1 torchvision == 0.15.2 timm == 0.9.2 pandas numpy