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