keras 如何抑制Tensorflow的所有签名警告?

9lowa7mx  于 2023-02-04  发布在  其他
关注(0)|答案(1)|浏览(125)

我收到了一个警告,但我找不到解决办法。显然是这样的:

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # or any {'0', '1', '2'}
import tensorflow as tf
os.environ['AUTOGRAPH_VERBOSITY'] = '1'

不足以阻止这条烦人的信息:

WARNING:tensorflow:AutoGraph could not transform <function Model.make_train_function.<locals>.train_function at 0x00000281A55264C8> and will run it as-is.
Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.
Cause: Bad argument number for Name: 4, expecting 3
To silence this warning, decorate the function with @tf.autograph.experimental.do_not_convert

它在每一个历元之后弹出。
是的,我已经使用@tf.autograph.experimental.do_not_convert修饰了我的每个函数,消息仍然弹出。我目前使用的是Tensorflow 2.2.0
如果你知道这件事发生的原因,那就太好了。我不知道“名字”是什么,因为在我的任何文件中都没有这样的名字。

hs1rzwqc

hs1rzwqc1#

试试这个:

tf.autograph.set_verbosity(0)

如果这不起作用,也许logging模块可以在这里提供帮助:

import logging
logging.getLogger("tensorflow").setLevel(logging.ERROR) # this goes *before* tf import
import tensorflow as tf

感谢@Shahrokh Bah添加该评论。

相关问题