torch~=1.7.1
tqdm~=4.58.0
numpy~=1.20.1
pytorch-lightning==1.1.7
wandb~=0.10.20
hydra-core~=1.0.6
