Upperbound the torchmetrics version (NVIDIA#5537)

Signed-off-by: andrusenkoau <[email protected]>
andrusenkoau · Jan 5, 2023 · 9eb267b · 9eb267b
1 parent 679825d
commit 9eb267b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt
@@ -2,7 +2,7 @@ hydra-core>=1.2.0,<1.3
 omegaconf>=2.2,<2.3
 pytorch-lightning>=1.8.3
 pyyaml<6  # Pinned until omegaconf works with pyyaml>=6
-torchmetrics>=0.4.1rc0
+torchmetrics>=0.4.1rc0,<=0.10.3
 transformers>=4.0.1,<=4.21.2
 wandb
 webdataset>=0.1.48,<=0.1.62