torch
python-dotenv
timm
torchvision