**顶!想问在Resnet网络结构中也输出所有层次的这四种信息,代码该怎么改
来源:5-5 Tensorboard实战(2)
战战的坚果
2020-03-29
def variable_summary(var, name):
with tf.name_scope(name):
mean = tf.reduce_mean(var)
with tf.name_scope(‘stddev’):
stddev = tf.sqrt(tf.reduce_mean(tf.square(var - mean)))
tf.summary.scalar(‘mean’, mean)
tf.summary.scalar(‘stddev’, stddev)
tf.summary.scalar(‘min’, tf.reduce_min(var))
tf.summary.scalar(‘max’, tf.reduce_max(var))
tf.summary.histogram(‘histogram’, var)
with tf.name_scope(‘summary’):
variable_summary(conv0, ‘conv0’)
variable_summary(conv0_0, ‘conv0_0’)
variable_summary(conv0_1, ‘conv0_1’)
variable_summary(conv1_0, ‘conv1_0’)
variable_summary(conv1_1, ‘conv1_1’)
variable_summary(conv1_2, ‘conv1_2’)
variable_summary(conv2_0, ‘conv2_0’)
variable_summary(conv2_1, ‘conv2_1’)
老师,我是直接粘的这段代码,如果还是没有缩进,我只能感到很抱歉,您也不用细看,因为这就是您课上讲的代码之一,我只是想问在Resnet网络结构中也输出所有层次的这四种信息,但是这段代码我不会改,我改成了下面的样子,就一直报错,找不到conv0和conv,想请问老师,该怎么改代码?我在inceptionnet和mobilebet上改的是可以成功运行的,只有resnet和它们都不一样。
with tf.name_scope(‘summary’):
variable_summary(conv0, ‘conv0’)
variable_summary(conv, ‘conv’)
ps,附截图,在我这边真的有缩进
1回答
-
正十七
2020-03-31
不知道你这段代码具体加在resnet的什么位置,但看问题的原因应该是conv0对这段代码是不可见的,你可以考虑用一个字典把每一层都保存下来,格式为{层的名字: 具体的一层}, 然后在构建好网络后遍历这个字典,对每个key:value对都去调用那个variable_summary的方法。
00
相似问题