torch
torchvision
timm == 0.4.5