From 75e3f9288eab9d384ed5cbf0e44e0d44f23d4ac9 Mon Sep 17 00:00:00 2001 From: Thomas O'Malley Date: Wed, 17 Jul 2019 12:36:22 -0700 Subject: [PATCH] Have `model_to_estimator` make a V2 Estimator if TF2 is enabled. PiperOrigin-RevId: 258620905 --- tensorflow/python/keras/estimator/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/python/keras/estimator/__init__.py b/tensorflow/python/keras/estimator/__init__.py index 79916aa6fc5..65a0e0c93ff 100644 --- a/tensorflow/python/keras/estimator/__init__.py +++ b/tensorflow/python/keras/estimator/__init__.py @@ -103,7 +103,8 @@ def model_to_estimator( custom_objects=custom_objects, model_dir=model_dir, config=config, - checkpoint_format=checkpoint_format) + checkpoint_format=checkpoint_format, + use_v2_estimator=False) @keras_export('keras.estimator.model_to_estimator', v1=[]) @@ -161,7 +162,6 @@ def model_to_estimator_v2( custom_objects=custom_objects, model_dir=model_dir, config=config, - checkpoint_format=checkpoint_format) + checkpoint_format=checkpoint_format, + use_v2_estimator=True) # LINT.ThenChange(//tensorflow_estimator/python/estimator/keras.py) - - -- GitLab