From a630db2c49d1ab8af82902d26894a3592139e520 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 2 Feb 2023 16:17:52 +0000 Subject: [PATCH] dlr eo: fixup --- aimodel/src/encoderonly_test_rainfall.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/encoderonly_test_rainfall.py b/aimodel/src/encoderonly_test_rainfall.py index c5896b2..6e6360b 100755 --- a/aimodel/src/encoderonly_test_rainfall.py +++ b/aimodel/src/encoderonly_test_rainfall.py @@ -46,7 +46,7 @@ if not os.path.exists(DIRPATH_OUTPUT): os.makedirs(os.path.join(DIRPATH_OUTPUT, "checkpoints")) if MIXED_PRECISION: - tf.keras.mixed_precision.set_policy("mixed_float16") + tf.keras.mixed_precision.set_global_policy("mixed_float16") # ██████ █████ ████████ █████ ███████ ███████ ████████