Upgrade dependency to torchmetrics == 1.0.1 (#205)

Author: mzweilin

pyproject.toml

@@ -15,7 +15,7 @@ dependencies = [
   "torchvision ~= 0.15.2",
   "lightning[extra] ~= 2.0.5", # Full functionality including TensorboardX.
   "pydantic == 1.10.11", # https://github.com/Lightning-AI/lightning/pull/18022/files
-  "torchmetrics == 1.0.0",
+  "torchmetrics == 1.0.1",
   "numpy == 1.23.5", # https://github.com/pytorch/pytorch/issues/91516
 
   # --------- hydra --------- #