torch>=1.8
pytorch-lightning>=1.6
torchmetrics>=0.8
matplotlib>=3.3
pyyaml>=5.4
