Skip to content

Commit 2ccb46a

Browse files
committed
fix name
1 parent eda6c82 commit 2ccb46a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

efficientnetv2/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -411,7 +411,7 @@ def set_precision_policy(policy_name=None):
411411
logging.info('use mixed precision policy name %s', policy_name)
412412
tf.compat.v1.keras.layers.enable_v2_dtype_behavior()
413413
policy = tf.keras.mixed_precision.Policy(policy_name)
414-
tf.keras.mixed_precision.set_policy(policy)
414+
tf.keras.mixed_precision.set_global_policy(policy)
415415

416416

417417
def build_model_with_precision(pp, mm, ii, tt, *args, **kwargs):

0 commit comments

Comments
 (0)