torch>=1.0
torchmetrics==0.3.2
