From 0a11de3451a9810feb583d1b8bad4600d883db76 Mon Sep 17 00:00:00 2001 From: valhassan Date: Wed, 16 Oct 2024 13:42:41 -0400 Subject: [PATCH] Refactor model script preprocessing method in script_model.py --- geo_deep_learning/tools/script_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/geo_deep_learning/tools/script_model.py b/geo_deep_learning/tools/script_model.py index e79275be..985a78c9 100644 --- a/geo_deep_learning/tools/script_model.py +++ b/geo_deep_learning/tools/script_model.py @@ -9,7 +9,7 @@ def __init__(self, model, data_module): def forward(self, x): sample = {"image": x} - preprocessed = self.data_module.model_script_preprocess(sample) + preprocessed = self.data_module._model_script_preprocess(sample) return self.model(preprocessed["image"]) def script_model(model, data_module):