From a3c9416cf0112867a0870d0677848a272801b68c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 8 Dec 2022 16:57:11 +0000 Subject: [PATCH] LossCrossentropy: don't sum --- aimodel/src/lib/ai/components/LossCrossentropy.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/aimodel/src/lib/ai/components/LossCrossentropy.py b/aimodel/src/lib/ai/components/LossCrossentropy.py index 99f81af..adc884e 100644 --- a/aimodel/src/lib/ai/components/LossCrossentropy.py +++ b/aimodel/src/lib/ai/components/LossCrossentropy.py @@ -16,10 +16,10 @@ 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) - label_nowater = tf.math.reduce_sum(tf.argmax(y_true, axis=-1)) + # result_reduce = tf.math.reduce_sum(result) + # label_nowater = tf.math.reduce_sum(tf.argmax(y_true, axis=-1)) # tf.print("DEBUG:TFPRINT:loss LABEL", y_true.shape, y_true, "LABEL_ARGMAX_COUNT_AXIS0", label_nowater, "PREDICT", y_pred.shape, y_pred, "BEFORE_REDUCE", result.shape, result, "AFTER_REDUCE", result_reduce.shape, result_reduce) - return result_reduce + return result def get_config(self): config = super(LossCrossentropy, self).get_config()