pytorch-lightning==1.3.5
torchmetrics==0.3.2
hydra-core==1.1.0
hydra-submitit-launcher==1.1.5.dev2
wandb==0.10.33
