torch>=1.6
torchmetrics>=0.2.0
pytorch-lightning>=1.1.1