even moar debug logging

This commit is contained in:
Starbeamrainbowlabs 2022-09-01 17:49:51 +01:00
parent 2663123407
commit e2f621e212
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -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))