将TensorFlow 1.x.x模型加载到TensorFlow 2.x.x中。

Question

我有一个使用TF1创建的SavedModel，现在想要在TF2中加载。

似乎每个图变量都会出现如下警告：

WARNING:tensorflow:Unable to create a python object for variable <tf.Variable 'Encoder_en/hidden_layers/tanh_layer_0/bias:0' shape=(512,) dtype=float32_ref> because it is a reference variable. It may not be visible to training APIs. If this is a problem, consider rebuilding the SavedModel after running tf.compat.v1.enable_resource_variables().

我希望你能尽可能修复这个警告或者只是禁止它的显示！

到目前为止，我已经尝试了：

# In my python app
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

# In my Dockerfile
ENV TF_CPP_MIN_LOG_LEVEL 2

编辑：这个模型来自于TensorFlow Hub，因此我没有构建它。

- dendog

Answer 1

TensorFlow最近版本中的登录方式发生了变化，不再使用TF_CPP_MIN_LOG_LEVEL（参见问题#26348和#31870）。请尝试使用tf.get_logger().setLevel('ERROR')。

import tensorflow as tf
tf.get_logger().warning('test')
# WARNING:tensorflow:test
tf.get_logger().setLevel('ERROR')
tf.get_logger().warning('test')
# (silence)

- jdehesa