torch>=1.9.0
pytorch-lightning~=1.4.2
torchmetrics~=0.5.0
tqdm~=4.62.1
wandb~=0.12.0
omegaconf~=2.1.1
commode-utils>=0.3.8
