hydra-core
omegaconf
pytorch-lightning~=1.3.8
scikit-learn
tensorboard
torch>=1.6.0
torchvision>=0.7.0
tqdm
