【发布时间】:2016-07-05 15:31:42
【问题描述】:
如何向张量板添加验证?我已经为图层编写了一个包装器,例如:
def convolution(input_data, kernel_shape, strides, activation, name=None):
with tf.name_scope(name):
kernel = tf.Variable(tf.truncated_normal(kernel_shape, stddev=stddev), name="weights")
bias = tf.Variable(tf.zeros([kernel_shape[-1]]), name="biases")
conv = tf.nn.conv2d(input=input_data, filter=kernel, strides=strides, padding="SAME", name="convolutions")
result = activation(tf.nn.bias_add(conv, bias), name="activations")
tf.scalar_summary(name + "/mean", tf.reduce_mean(kernel))
return result
并在main 中使用summary_op = tf.merge_all_summaries()。我还实现了train_op 和valid_op,它们都调用inference 函数。但是,出现了一个错误,即我们有重复的 scalar_summary 标签,即inference 用于train_op 和valid_op,这会导致重复,例如conv1/mean 摘要。
我怎样才能做到这一点?我需要的是使用相同的函数inference 运行训练和验证。
【问题讨论】: