torch>=1.8
pytorch-lightning>=1.7
torchmetrics>=0.9
matplotlib>=3.3
pyyaml>=5.4
