首页 > 解决方案 > Keras 中的 tensorflow 会话在哪里

问题描述

我是 keras 和 tensorflow 的新手。当我用 tensorflow 编写程序时,我必须建立一个会话来运行图形。但是,我在使用keras的时候,虽然后端明显是tensorflow,但是在keras代码中并没有看到session。似乎所有事情都是在model.compile和model.fit之后完成的。

那么,Keras 是如何工作的呢?张量流会话在哪里?而不是会话,我可以使用 keras 的急切执行吗?

提前感谢,对不起我的英语

标签: pythontensorflowkerasdeep-learningkeras-layer

解决方案


Keras 没有直接的会话,因为它支持多个后端。假设您使用 TF 作为后端,您可以获得全局会话:

from keras import backend as K
sess = K.get_session()

另一方面,如果您已经打开Session并希望将其设置为 Keras 应该使用的会话,您可以通过以下方式进行:

K.set_session(sess)

推荐阅读