pytorch_lightning==1.5.10
torchvision==0.11.1
torchmetrics==0.7.2
omegaconf==2.1.1
hydra_core==1.1.1
craterdata==0.1.3
python-dotenv==0.19.2
rich==11.2.0
hydra_colorlog
hydra-optuna-sweeper
