torch
torchvision
pillow
numpy
einops
thop
timm
toml
