pytorch_lightning==1.5.0
PyYAML==6.0
wandb==0.12.7
