Flexible version for torch

Joao-L-S-Almeida · Joao-L-S-Almeida · commit 59a71bab0b92 · 2024-07-08T17:22:12.000-03:00
Signed-off-by: João Lucas de Sousa Almeida &lt;joao.l.sa.9.3@gmail.com&gt;
diff --git a/requirements/required.txt b/requirements/required.txt
@@ -2,12 +2,12 @@ torchgeo==0.5.1
 rioxarray==0.15.0
 albumentations==1.3.1
 rasterio==1.3.9
-torch==2.1.0
+torch>=2.1.0
 torchvision==0.16.0
 torchmetrics==1.3.1
 geopandas==0.14.2
 lightly==1.4.25
 h5py==3.10.0
 geobench==1.0.0
 mlflow==2.12.1
-lightning==2.2.5
+lightning==2.2.5
