From 3e42972fb0eba31fb299bdc0f5c6fde4524682ed Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 11 May 2023 17:45:52 +0100 Subject: [PATCH] dlr: add comment --- 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 cf19655..fdbd1f3 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -46,7 +46,7 @@ PATH_COLOURMAP = os.environ["PATH_COLOURMAP"] STEPS_PER_EPOCH = int(os.environ["STEPS_PER_EPOCH"]) if "STEPS_PER_EPOCH" in os.environ else None REMOVE_ISOLATED_PIXELS = False if "NO_REMOVE_ISOLATED_PIXELS" in os.environ else True EPOCHS = int(os.environ["EPOCHS"]) if "EPOCHS" in os.environ else 50 -LOSS = os.environ["LOSS"] if "LOSS" in os.environ else "cross-entropy-dice" +LOSS = os.environ["LOSS"] if "LOSS" in os.environ else "cross-entropy-dice" # other possible valuesL cross-entropy DICE_LOG_COSH = True if "DICE_LOG_COSH" in os.environ else False LEARNING_RATE = float(os.environ["LEARNING_RATE"]) if "LEARNING_RATE" in os.environ else 0.001 WATER_THRESHOLD = float(os.environ["WATER_THRESHOLD"]) if "WATER_THRESHOLD" in os.environ else 0.1