Tensorflow SavedModelBuilder bug 解决

Hijiao

于 2018-04-20 13:21:38 发布

阅读量3.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow

本文链接：https://blog.csdn.net/fo11ower/article/details/80017454

本文介绍了在使用Tensorflow的SavedModelBuilder进行模型导出和部署时遇到的问题，特别是当模型从本地迁移至Tensorflow Serving时出现的不兼容性问题。通过分析，确定问题是由于Tensorflow版本差异及GPU支持导致。解决方案是在导出模型时添加strip_default_attrs=True参数，以确保在不同环境中加载模型的兼容性。参考了官方文档和GitHub上的相关issue。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

之前训练文件的导出和freeze都是使用tf.train.Saver()方法,这次为了适配 Tensorflow Serving 使用了tf.saved_model.builder.SavedModelBuilder()的方法. 经过一天的尝试,解决了 builder的save和load.在本机导出再载入做inference没问题,但是将模型部署到serving下就碰到了如下问题(粘贴为引用格式太乱了,就粘贴为代码块啦…):

Traceback (most recent call last):
  File "/home/max/code/tf-serving/serving/tensorflow_serving/example/inception_client_gold.py", line 64, in <module>
    tf.app.run()
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 126, in run
    _sys.exit(main(argv))
  File "/home/max/code/tf-serving/serving/tensorflow_serv