torch>=1.8
pytorch-lightning>=1.6
torchmetrics>=0.6
matplotlib>=3.3
