torch>1.7
pytorch-lightning==1.4.5
torchmetrics>0.5
matplotlib>=3.3
