Skip to content

Commit ceb8954

Browse files
author
Darius Couchard
committed
Keep batch_size at 4096 for the moment
2 parents f407e47 + 679f33c commit ceb8954

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/worldcereal/openeo/feature_extractor.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ class PrestoFeatureExtractor(PatchFeatureExtractor):
1919
"""
2020

2121
PRESTO_MODEL_URL = "https://artifactory.vgt.vito.be/artifactory/auxdata-public/worldcereal-minimal-inference/presto.pt" # NOQA
22-
PRESO_WHL_URL = "https://artifactory.vgt.vito.be/artifactory/auxdata-public/worldcereal/dependencies/presto_worldcereal-0.1.0-py3-none-any.whl"
22+
PRESO_WHL_URL = "https://artifactory.vgt.vito.be/artifactory/auxdata-public/worldcereal/dependencies/presto_worldcereal-0.1.1-py3-none-any.whl"
2323
BASE_URL = "https://s3.waw3-1.cloudferro.com/swift/v1/project_dependencies" # NOQA
2424
DEPENDENCY_NAME = "worldcereal_deps.zip"
2525

@@ -104,7 +104,11 @@ def execute(self, inarr: xr.DataArray) -> xr.DataArray:
104104

105105
self.logger.info("Extracting presto features")
106106
features = get_presto_features(
107+
<<<<<<< HEAD
107108
inarr, presto_model_url, self.epsg, batch_size=8192
109+
=======
110+
inarr, presto_model_url, self.epsg, batch_size=4096
111+
>>>>>>> main
108112
)
109113
return features
110114

0 commit comments

Comments
 (0)