解决tensorflow+gunicorn+flask的flag异常 UnrecognizedFlagError: Unknown command line flag

最新推荐文章于 2023-10-23 10:51:55 发布

wangyueshu

最新推荐文章于 2023-10-23 10:51:55 发布

阅读量1.3k

点赞数

分类专栏：模型文章标签： tensorflow gunicorn flask

本文链接：https://blog.csdn.net/wangyueshu/article/details/107039536

版权

在使用TensorFlow、Gunicorn和Flask搭建模型预测并发API生产环境中遇到UnrecognizedFlagError异常。问题源于gunicorn启动命令行参数被TensorFlow解析导致。解决方案包括修改TensorFlow的flags.py代码或工程模型代码，避免TensorFlow尝试解析不相关参数。此外，文章还介绍了如何定位和解决问题，包括模型实例化的正确位置以及查看错误信息的方法。

摘要由CSDN通过智能技术生成

背景：TextCnn模型用tensorflow+flask+gunicorn搭建模型预测并发API生产环境，模型调用抛异常。

报错代码：

#模型代码处：

# Misc Parameters
tf.flags.DEFINE_boolean("allow_soft_placement", True, "Allow device soft device placement")
tf.flags.DEFINE_boolean("log_device_placement", False, "Log placement of ops on devices")

FLAGS = tf.flags.FLAGS
FLAGS.flag_values_dict()

FLAGS.set_default("checkpoint_dir",checkpoint_dir)

# Map data into vocabulary
vocab_path = os.path.join(FLAGS.checkpoint_dir, "..", "vocab")
self.vocab_processor = learn.preprocessing.VocabularyProcessor.restore(vocab_path)

#tensorflow的flags.py代码处：
class _FlagValuesWrapper(object):
  """Wrapper class for absl.flags.FLAGS.

  The difference is that tf.flags.FLAGS implicitly parses flags with sys.argv
  whe