导入keras时,将显示“使用XXX后端”。
import keras
Run Code Online (Sandbox Code Playgroud)
输出:
Using TensorFlow backend.
Run Code Online (Sandbox Code Playgroud)
但是我清楚知道我在用什么。
如何禁用它?
只需使用下面的代码。
import os
import sys
stderr = sys.stderr
sys.stderr = open(os.devnull, 'w')
import keras
sys.stderr = stderr
Run Code Online (Sandbox Code Playgroud)
您有两个选择:首先在此处注释keras/backend/__init__.pystring print('Using TensorFlow backend.'),但当然不建议使用此选项,因为您将在 keras 模块中编辑代码。第二个选项是:
import sys
stdout = sys.stdout
sys.stdout = open('/dev/null', 'w')
import keras
sys.stdout = stdout
Run Code Online (Sandbox Code Playgroud)
它并不优雅,但很有效。
| 归档时间: |
|
| 查看次数: |
3944 次 |
| 最近记录: |