torch>1.7
pytorch-lightning>1.3
torchmetrics>0.3
matplotlib>=3.3
