From ccd256c00a01322c582a17417d5812371c95dbd0 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 15 Sep 2022 17:37:04 +0100 Subject: [PATCH] embed rainfall radar, not both --- 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 fbde335..18943e6 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.predict_on_batch(batch) # ((rainfall, water), dummy_label) + rainfall = self.model_predict.predict_on_batch(batch[0]) # ((rainfall, water), dummy_label) for step in tf.unstack(rainfall, axis=0): yield step