torch>=1.0
torchmetrics>=0.10.0
