torch>=1.7
torchmetrics>=0.10.0
