将 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 许可协议

阅读 641
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 许可协议