From bb0258f5cde2963ec0cb0d52d51330c16b4ec9d5 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 21 Oct 2022 15:38:57 +0100 Subject: [PATCH] flip squeeze operator ordering --- aimodel/src/subcommands/train_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_predict.py b/aimodel/src/subcommands/train_predict.py index 0d2b8ba..fb94adb 100644 --- a/aimodel/src/subcommands/train_predict.py +++ b/aimodel/src/subcommands/train_predict.py @@ -99,7 +99,7 @@ def do_png(args, ai, dataset, model_code): rainfall, water = item water_predict_batch = ai.embed(rainfall) - water = tf.unstack(tf.squeeze(water), axis=0) + water = tf.squeeze(tf.unstack(water, axis=0)) i_batch = 0 for water_predict in water_predict_batch: