From aa3831bc4f3c3982f54d1a4bd9978358a3e6e851 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 17 Jan 2023 19:07:16 +0000 Subject: [PATCH] ds_eo: pass water to reshape --- aimodel/src/lib/dataset/dataset_encoderonly.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset_encoderonly.py b/aimodel/src/lib/dataset/dataset_encoderonly.py index ea8ae54..e50aea9 100644 --- a/aimodel/src/lib/dataset/dataset_encoderonly.py +++ b/aimodel/src/lib/dataset/dataset_encoderonly.py @@ -88,7 +88,7 @@ def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rain ) rainfall = tf.reshape(rainfall, [-1, windowsize, windowsize, rainfall_channels]) - water = tf.reshape([-1]) # we flatten because we cropped to the right shape above + water = tf.reshape(water, [-1]) # we flatten because we cropped to the right shape above print("DEBUG DATASET_OUT:rainfall shape", rainfall.shape) print("DEBUG DATASET_OUT:water shape", water.shape)