From b6ca885fe274b834d861893a49628f693869e5e0 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 9 Mar 2023 19:43:35 +0000 Subject: [PATCH] dlr: told you so --- 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 666e14f..24d4f5a 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -164,7 +164,7 @@ else: # Tell Tensorflow about our custom layers so that it can deserialise models that use them "LossCrossEntropyDice": LossCrossEntropyDice, "metric_dice_coefficient": dice_coefficient, - "make_sensitivity": sensitivity, + "sensitivity": sensitivity, "specificity": specificity, "one_hot_mean_iou": mean_iou })