tf.ConfigProto()函数
发布日期:2021-07-01 04:20:45 浏览次数:2 分类:技术文章

本文共 240 字,大约阅读时间需要 1 分钟。

tf.ConfigProto一般用在创建session的时候用来对session进行参数配置。

tf.ConfigProto()的参数

  • log_device_placement = True : 是否打印设备分配日志
  • allow_soft_placement = True : 如果你指定的设备不存在,允许TF自动分配设备

示例代码:

tf.ConfigProto(log_device_placement = True, allow_soft_placement = True)

转载地址:https://mtyjkh.blog.csdn.net/article/details/114887718 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:tf.train.MomentumOptimizer()优化器
下一篇:tensorflow2中如何设置GPU的使用

发表评论

最新留言

哈哈,博客排版真的漂亮呢~
[***.90.31.176]2024年05月07日 20时51分33秒