将 TensorFlow 设置为 Keras 中的会话的问题

新手上路,请多包涵

执行以下操作:

 from keras import backend as K
sess = tf.Session()
K.set_session(sess)

即使我已经正确导入了 Keras 和 TensorFlow,我还是得到以下信息:

 module 'keras.backend' has no attribute 'set_session'

任何帮助,将不胜感激!

原文由 MaxR 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 548
1 个回答

需要 .compat.v1.tf —之后

例子:

 tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config));

(在 Anaconda、Python 3.7、tensorflow 2.0.0 中测试)

原文由 leenremm 发布,翻译遵循 CC BY-SA 4.0 许可协议

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进