From 2ccb46aa1c0c79a5ca0645c8e2b5318cf2e9fdfe Mon Sep 17 00:00:00 2001 From: fsx950223 Date: Tue, 3 May 2022 16:27:19 +0800 Subject: [PATCH] fix name --- efficientnetv2/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/efficientnetv2/utils.py b/efficientnetv2/utils.py index 3727ba1ca..039ca65a1 100644 --- a/efficientnetv2/utils.py +++ b/efficientnetv2/utils.py @@ -411,7 +411,7 @@ def set_precision_policy(policy_name=None): logging.info('use mixed precision policy name %s', policy_name) tf.compat.v1.keras.layers.enable_v2_dtype_behavior() policy = tf.keras.mixed_precision.Policy(policy_name) - tf.keras.mixed_precision.set_policy(policy) + tf.keras.mixed_precision.set_global_policy(policy) def build_model_with_precision(pp, mm, ii, tt, *args, **kwargs):