Fix torchmetrics requirement

LukasHedegaard · May 19, 2022 · e1dcdf9 · e1dcdf9
1 parent 40ffe42
commit e1dcdf9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 corider
 pytorch-lightning>=1.3,<1.5
-torchmetrics
+torchmetrics>=0.4,<0.8
 python-dotenv
 supers>=0.2  
 matplotlib