TensorFlow:查询会话配置

时间:2019-01-14 19:54:05

标签: tensorflow

是否可以查询TensorFlow会话使用的当前配置?例如

config = tf.ConfigProto()
config.intra_op_parallelism_threads = 44
config.inter_op_parallelism_threads = 44
sess = tf.Session(config=config)

在TensorFlow(我正在使用的版本1.10)中,我似乎找不到一种方法来读取sess变量中的配置值。

1 个答案:

答案 0 :(得分:0)

您可以通过以下方式访问“私有”变量_config

sess._config

这是类型tensorflow.core.protobuf.config_pb2.ConfigProto的类(简而言之,您可以访问intra_op_parallelism_threadsinter_op_parallelism_threads以及在tf.ConfigProto定义中添加的任何其他属性)。