tqdm>=4.49.0
torchmetrics>=0.4.1
torch>=1.9.0
omegaconf>=2.1.0
pytorch_lightning>=1.4.0
