torch>=1.7.1
torchvision>=0.8.2
torchmetrics>=0.2.0
