diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index b89597c..fdb2b12 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -1,9 +1,11 @@
 torch>=2.0.0
 torchvision>=0.15
 lightning
+datasets
 rasterio
 rioxarray
 xarray
+geopandas
 tifffile
 webcolors
 tiler
@@ -20,3 +22,5 @@ joblib
 GDAL>=3.3.0
 coverage
 deepspeed
+webdataset
+torchgeo