tf.ConfigProto()函数
生活随笔
收集整理的這篇文章主要介紹了
tf.ConfigProto()函数
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
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)總結
以上是生活随笔為你收集整理的tf.ConfigProto()函数的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: tensorflow2中如何设置GPU的
- 下一篇: tf.train.MomentumOpt