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一般用在创建session的时候用来对session进行参数配置。
tf.ConfigProto()的参数
示例代码:
tf.ConfigProto(log_device_placement = True, allow_soft_placement = True)