Update torchmetrics requirement from <1.0,>=0.9 to >=0.9,<2.0

Update torchmetrics requirement from <1.0,>=0.9 to >=0.9,<2.0 #362