From e2f621e2122b24ac07058e58231fd4fba2bb3336 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 1 Sep 2022 17:49:51 +0100 Subject: [PATCH] even moar debug logging --- aimodel/src/lib/dataset/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset.py b/aimodel/src/lib/dataset/dataset.py index e2b9a44..cd7f26a 100644 --- a/aimodel/src/lib/dataset/dataset.py +++ b/aimodel/src/lib/dataset/dataset.py @@ -25,7 +25,7 @@ def parse_item(item): water = tf.expand_dims(water, axis=-1) # TODO: The shape of the resulting tensor can't be statically determined, so we need to reshape here - + print("DEBUG:dataset ITEM rainfall:shape", rainfall.shape, "water:shape", water.shape) # TODO: Any other additional parsing here, since multiple .map() calls are not optimal return ((rainfall, water), tf.ones(1))