From 82e01da70bc86acb6d5bcd2f07296f5899c37c89 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 13 Jan 2023 18:47:29 +0000 Subject: [PATCH] fixup.... again oops --- aimodel/src/deeplabv3_plus_test_rainfall.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index b9f1d88..1d733cd 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -161,7 +161,7 @@ if PATH_CHECKPOINT is None: if LOSS == "cross-entropy-dice": loss_fn = LossCrossEntropyDice() elif LOSS == "cross-entropy": - tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) + loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) else: raise Exception(f"Error: Unknown loss function '{LOSS}' (possible values: cross-entropy, cross-entropy-dice).")