get_variables_path fix

alibaba · Oct 14, 2024 · 8df7de2 · 8df7de2
1 parent 6483f7a
commit 8df7de2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easy_rec/python/tools/split_model_pai.py b/easy_rec/python/tools/split_model_pai.py
@@ -17,7 +17,7 @@
   from tensorflow.python.saved_model.path_helpers import get_variables_path
   from tensorflow.python.ops.resource_variable_ops import _from_proto_fn
 else:
-  from tensorflow.compat.v1.python.saved_model.utils_impl import get_variables_path
+  from tensorflow.python.saved_model.utils_impl import get_variables_path
 
 FLAGS = tf.app.flags.FLAGS
 tf.app.flags.DEFINE_string('model_dir', '', '')