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()