From 8770638022f0b7ad442c1644624e1a26c710050e Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 14 Sep 2022 17:41:01 +0100 Subject: [PATCH] ai: call .predict(), not the model itself --- aimodel/src/lib/ai/RainfallWaterContraster.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py index 2ab91fe..39b2db8 100644 --- a/aimodel/src/lib/ai/RainfallWaterContraster.py +++ b/aimodel/src/lib/ai/RainfallWaterContraster.py @@ -88,7 +88,7 @@ class RainfallWaterContraster(object): i_batch = -1 for batch in dataset: i_batch += 1 - rainfall = self.model_predict(batch) # ((rainfall, water), dummy_label) + rainfall = self.model_predict.predict(batch) # ((rainfall, water), dummy_label) for step in tf.unstack(rainfall, axis=0): yield step