From d55a13f536685d6a6dc33e0ccd0d7c4302143b1b Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 24 Nov 2022 18:24:03 +0000 Subject: [PATCH] debug --- aimodel/src/subcommands/train_mono_predict.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/aimodel/src/subcommands/train_mono_predict.py b/aimodel/src/subcommands/train_mono_predict.py index 4e3c5c1..ab0cd44 100644 --- a/aimodel/src/subcommands/train_mono_predict.py +++ b/aimodel/src/subcommands/train_mono_predict.py @@ -149,7 +149,10 @@ def do_jsonl(args, ai, dataset, model_params, do_argmax=False): for batch in batched_iterator(dataset, tensors_in_item=2, batch_size=model_params["batch_size"]): rainfall_actual_batch, water_actual_batch = batch + print("DEBUG:do_jsonl rainfall_actual_batch", rainfall_actual_batch.shape) + print("DEBUG:do_jsonl water_actual_batch", water_actual_batch.shape) water_predict_batch = ai.embed(rainfall_actual_batch) + print("DEBUG:do_jsonl water_actual_batch", water_predict_batch.shape) water_actual_batch = tf.unstack(water_actual_batch, axis=0) rainfall_actual_batch = tf.unstack(rainfall_actual_batch, axis=0)