hydra-core
omegaconf
pytorch-lightning<1.5,>=1.3.8
scikit-learn
tensorboard
torch>=1.8.1
torchvision>=0.9.1
tqdm
