From 0f0b691b5d760882da80ea89e01910db94a4aa0d Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 13 Jan 2023 18:09:31 +0000 Subject: [PATCH] loss cel + dice: fixup --- aimodel/src/lib/ai/components/LossCrossEntropyDice.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/ai/components/LossCrossEntropyDice.py b/aimodel/src/lib/ai/components/LossCrossEntropyDice.py index 7e7713a..ae07525 100644 --- a/aimodel/src/lib/ai/components/LossCrossEntropyDice.py +++ b/aimodel/src/lib/ai/components/LossCrossEntropyDice.py @@ -26,7 +26,7 @@ class LossCrossEntropyDice(tf.keras.losses.Loss): """ def __init__(self, **kwargs): - super(LossDice, self).__init__(**kwargs) + super(LossCrossEntropyDice, self).__init__(**kwargs) def call(self, y_true, y_pred): y_true = tf.cast(y_true, tf.float32) @@ -34,7 +34,7 @@ class LossCrossEntropyDice(tf.keras.losses.Loss): return tf.reduce_mean(o) def get_config(self): - config = super(LossDice, self).get_config() + config = super(LossCrossEntropyDice, self).get_config() config.update({ })