pytorch>=1.5
pytorch-lightning>=0.9.0
pytorch-lightning-bolts>=0.2.2
hydra-core>=1.0.0
scikit-learn>=0.23.1
# Only necessary for distributed training:
# ray[tune]
# munch
# wandb
