From 7dba03200f015626afc00bcc3f013fce0de86013 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 24 Nov 2022 19:06:48 +0000 Subject: [PATCH] fixup --- aimodel/src/subcommands/rainfall_stats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/rainfall_stats.py b/aimodel/src/subcommands/rainfall_stats.py index e332eef..f68492d 100755 --- a/aimodel/src/subcommands/rainfall_stats.py +++ b/aimodel/src/subcommands/rainfall_stats.py @@ -61,7 +61,7 @@ def run(args): batch_stddev = tf.math.reduce_std(rainfall_flat) batch_max = tf.math.reduce_max(rainfall_flat) - print("BATCH mean\t", batch_mean.numpy().tolist()[0], "stddev\t", batch_stddev.numpy().tolist()[0], "max\t", batch_max.numpy().tolist()[0]) + print("BATCH mean\t", batch_mean.numpy().tolist(), "stddev\t", batch_stddev.numpy().tolist(), "max\t", batch_max.numpy().tolist()) calc_mean.append(batch_mean) calc_stddev.append(batch_stddev)