From 279e27c8986370c878fc42f1649252e3b1ba08fa Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 14 Sep 2022 17:16:49 +0100 Subject: [PATCH] fixup --- 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 0c3fdaa..a6d7890 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 - result_batch = self.model(batch[0]) # ((rainfall, water), dummy_label) + result_batch = self.model_predict(batch[0]) # ((rainfall, water), dummy_label) rainfall, water = tf.unstack(result_batch, axis=-2) rainfall = tf.unstack(rainfall, axis=0)