Flexible version for torchvision and torchmetrics

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Jul 8, 2024 · 2661c17 · 2661c17
1 parent 59a71ba
commit 2661c17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/required.txt b/requirements/required.txt
@@ -3,8 +3,8 @@ rioxarray==0.15.0
 albumentations==1.3.1
 rasterio==1.3.9
 torch>=2.1.0
-torchvision==0.16.0
-torchmetrics==1.3.1
+torchvision>=0.16.0
+torchmetrics>=1.3.1
 geopandas==0.14.2
 lightly==1.4.25
 h5py==3.10.0