torch
torchvision
timm==0.4.12
easydict
