torch>=1.12
torchvision
PyYAML
tqdm
omegaconf
scikit-learn
wandb
rich
