torch==2.4.0
torchvision
wandb