torch>=1.10.0
torchvision
hydra-core==1.2.0
wandb
omegaconf