hydra-core
omegaconf
torchmetrics==0.6.0
pytorch-lightning>=1.3.8,<1.5
scikit-learn
tensorboard
torch>=1.10.0
torchvision>=0.11.0
tqdm
h5py
