From 3131b4f7b39644ceb031b3183ea9fbcf4a0f66e4 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 24 Nov 2022 18:25:32 +0000 Subject: [PATCH] debug2 --- aimodel/src/subcommands/train_mono_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_mono_predict.py b/aimodel/src/subcommands/train_mono_predict.py index ab0cd44..c0e169f 100644 --- a/aimodel/src/subcommands/train_mono_predict.py +++ b/aimodel/src/subcommands/train_mono_predict.py @@ -152,12 +152,12 @@ def do_jsonl(args, ai, dataset, model_params, do_argmax=False): 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) i_batch = 0 for water_predict in water_predict_batch: + print("DEBUG:do_jsonl water_predict", water_predict.shape) # [ width, height, softmax_probabilities ] → [ batch, width, height ] if do_argmax: water_predict = tf.math.argmax(water_predict, axis=-1)