From 5b54ceec48b099a62d8f9df1cbcd611479a453dd Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 20 Jan 2023 18:36:14 +0000 Subject: [PATCH] ds eo: debug --- aimodel/src/lib/dataset/dataset_encoderonly.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aimodel/src/lib/dataset/dataset_encoderonly.py b/aimodel/src/lib/dataset/dataset_encoderonly.py index de5b16d..a529648 100644 --- a/aimodel/src/lib/dataset/dataset_encoderonly.py +++ b/aimodel/src/lib/dataset/dataset_encoderonly.py @@ -26,6 +26,7 @@ def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rain print("DEBUG DATASET:water shape", metadata["waterdepth"]) print("DEBUG DATASET:water_threshold", water_threshold) print("DEBUG DATASET:water_bins", water_bins) + print("DEBUG DATASET:rainfall_scale_up", rainfall_scale_up) if heightmap is not None: heightmap = tf.expand_dims(heightmap, axis=-1)