fix name

google · May 3, 2022 · 2ccb46a · 2ccb46a
1 parent eda6c82
commit 2ccb46a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):