TensorFlow ConfigProto

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

 

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

　　

1 session_conf = tf.ConfigProto(log_device_placement = True, allow_soft_placement = True)
2 　　sess = tf.Session(config = session_conf)
3 　　with sess.as_default():
4     　　pass

 

 

 

 

 

posted @ 2017-04-25 17:21  xingchao  阅读(264)  评论(0)    收藏  举报