From f48473b70327b754978e32c0e7030f22dd7e3a8b Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 28 Nov 2022 19:00:11 +0000 Subject: [PATCH] fixup --- aimodel/src/lib/ai/components/LossCrossentropy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/LossCrossentropy.py b/aimodel/src/lib/ai/components/LossCrossentropy.py index 46a6807..35c6ea5 100644 --- a/aimodel/src/lib/ai/components/LossCrossentropy.py +++ b/aimodel/src/lib/ai/components/LossCrossentropy.py @@ -17,7 +17,7 @@ class LossCrossentropy(tf.keras.losses.Loss): def call(self, y_true, y_pred): result = tf.keras.metrics.categorical_crossentropy(y_true, y_pred) result_reduce = tf.math.reduce_sum(result) - tf.print("DEBUG:TFPRINT:loss LABEL", y_true, "PREDICT", y_pred, "BEFORE_REDUCE", result.shape, result, "AFTER_REDUCE", result_reduce.shape, result_reduce) + tf.print("DEBUG:TFPRINT:loss LABEL", y_true.shape, y_true, "PREDICT", y_pred.shape, y_pred, "BEFORE_REDUCE", result.shape, result, "AFTER_REDUCE", result_reduce.shape, result_reduce) return result def get_config(self):