keras tensorflow session

红太狼 2021-12-18 16:51 364阅读 0赞

获取keras已经建立的session：

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

给keras设置传递一个建立好的session：

K.set_session(sess)

参考：

[https://stackoverflow.com/questions/53653303/where-is-the-tensorflow-session-in-keras][https_stackoverflow.com_questions_53653303_where-is-the-tensorflow-session-in-keras]

正常使用keras建立好的session：

传递给compile的\*\*kwargs将会被传递给session：

from tensorflow.core.protobuf import config_pb2
    with session.Session() as sess:
        sess.run(c, options=config_pb2.RunOptions(
            report_tensor_allocations_upon_oom=True))

可以被写成：

import tensorflow as tf
    run_opts = tf.RunOptions(report_tensor_allocations_upon_oom = True)
    
    model.compile(loss = "...", optimizer = "...", metrics = "..", options = run_opts)

参考：[https://github.com/tensorflow/tensorflow/issues/17076][https_github.com_tensorflow_tensorflow_issues_17076]

[https://stackoverflow.com/questions/49665757/how-to-add-report-tensor-allocations-upon-oom-to-runoptions-in-keras][https_stackoverflow.com_questions_49665757_how-to-add-report-tensor-allocations-upon-oom-to-runoptions-in-keras]

[https_stackoverflow.com_questions_53653303_where-is-the-tensorflow-session-in-keras]: https://stackoverflow.com/questions/53653303/where-is-the-tensorflow-session-in-keras
[https_github.com_tensorflow_tensorflow_issues_17076]: https://github.com/tensorflow/tensorflow/issues/17076
[https_stackoverflow.com_questions_49665757_how-to-add-report-tensor-allocations-upon-oom-to-runoptions-in-keras]: https://stackoverflow.com/questions/49665757/how-to-add-report-tensor-allocations-upon-oom-to-runoptions-in-keras

发表评论取消回复

表情：

评论列表（有 0 条评论，364人围观）

还没有评论，来说两句吧...

相关阅读

相关 anaconda 下安装tensorflow & keras

　　首先，同胞们要记住，你要做什么？该怎么做？你的目标是什么？千万不要因为中间遇到的连带问题，而忘记了你要做什么？一下开始介绍：　　? ? ? ? 下载：官网速度很慢，容易

青旅半醒/ 2023年09月24日 09:44/ 0 赞/ 182 阅读

相关 tensorflow_tf.Session()

Session 是 Tensorflow 为了控制,和输出文件执行的语句. 运行 session.run() 可以获得你要得知的运算结果, 或者是你所要运算的部分.

青旅半醒/ 2023年08月17日 15:58/ 0 赞/ 97 阅读

相关 TensorFlow和keras对应版本

<table> <thead> <tr> <th>Framework</th> <th>Env name (--env parameter)</th>

╰+攻爆jí腚メ/ 2023年02月25日 07:14/ 0 赞/ 15 阅读

相关 Keras tensorflow Resnet模型

Resnet相关文章整理 1、Keras大法（9）——实现ResNet-34模型 [https://blog.csdn.net/weixin\_42499236/a

小灰灰/ 2022年12月27日 07:26/ 0 赞/ 290 阅读

相关【Tensorflow+Keras】用Tensorflow.keras的方法替代keras.layers.merge

1 旧版在旧版的keras中，keras.layers.merge的使用方法 keras.layers.merge([layer1,layer2],

柔光的暖阳◎/ 2022年10月05日 14:56/ 0 赞/ 227 阅读

相关 tensorflow.keras metrics应用

tensorflow.keras metrics应用 > 不使用metrics实现的博客参考： ![image-20210804195738825][] im

冷不防/ 2022年09月08日 00:09/ 0 赞/ 180 阅读

相关 keras转tensorflow

参考：[https://juejin.im/post/5b7bccc6e51d453887102e0a][https_juejin.im_post_5b7bccc6e51d45

野性酷女/ 2022年04月10日 13:16/ 0 赞/ 246 阅读

相关 keras tensorflow session

获取keras已经建立的session： from keras import backend as K sess = K.get_session() 给ke

红太狼/ 2021年12月18日 16:51/ 0 赞/ 365 阅读

相关 TensorFlow和keras结合使用

喜欢keras风格的模型搭建而不喜欢tensorflow的方式。但是个人觉得tensorflow的对于loss function定义的灵活性，还是非常便捷的，所以秉承着将

布满荆棘的人生/ 2021年12月18日 07:45/ 0 赞/ 328 阅读

相关 Tensorflow-keras实战一

一、keras 1. keras是什么 ◆基于python的高级神经网络API，不是一个完整的库。 ◆Francois Chollet于2014-2C15年编

Bertha 。/ 2021年09月23日 04:52/ 0 赞/ 346 阅读