From a9b0246a5dd50dc8c4b51bea924ef2348dbe1254 Mon Sep 17 00:00:00 2001 From: Chen Ding Date: Fri, 11 Nov 2022 17:04:20 +0800 Subject: [PATCH 1/2] [Estimator] Fix Estimator export_savedmodel to oss. (#7) --- tensorflow_estimator/python/estimator/estimator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow_estimator/python/estimator/estimator.py b/tensorflow_estimator/python/estimator/estimator.py index 148a5d51..cecf5a49 100644 --- a/tensorflow_estimator/python/estimator/estimator.py +++ b/tensorflow_estimator/python/estimator/estimator.py @@ -880,7 +880,8 @@ def _export_all_saved_models( gfile.MakeDirs(dest_path) gfile.Copy(source, dest_absolute) - gfile.Rename(temp_export_dir, export_dir) + # '/' is needed when rename a oss directory. + gfile.Rename(temp_export_dir + '/', export_dir) return export_dir def _add_meta_graph_for_mode(self, From 4e100c76f5e9a6b84ea13f8a83354f3a0a630951 Mon Sep 17 00:00:00 2001 From: Chen Ding Date: Fri, 11 Nov 2022 17:04:20 +0800 Subject: [PATCH 2/2] [Estimator] Fix encode string to byte in python3. --- tensorflow_estimator/python/estimator/estimator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_estimator/python/estimator/estimator.py b/tensorflow_estimator/python/estimator/estimator.py index cecf5a49..79d44e89 100644 --- a/tensorflow_estimator/python/estimator/estimator.py +++ b/tensorflow_estimator/python/estimator/estimator.py @@ -881,7 +881,7 @@ def _export_all_saved_models( gfile.Copy(source, dest_absolute) # '/' is needed when rename a oss directory. - gfile.Rename(temp_export_dir + '/', export_dir) + gfile.Rename(temp_export_dir + b'/', export_dir) return export_dir def _add_meta_graph_for_mode(self,