您的位置:首页 > 其它

tensorsor快速获取所有变量,和快速计算L2范数

2017-11-02 10:46 211 查看
可以指定要的变量,计算L2范数。

for var in tf.trainable_variables():
print (var.name)
print (var.get_shape())
print (sess.run(tf.nn.l2_loss(var)))


for var in tf.trainable_variables():
if var.op.name.startswith("InceptionV4/Logits/Logits/weights"):
print (var)
print (var.get_shape())
print (sess.run(tf.nn.l2_loss(var)))

只求weights的L2范数。(直径正则化的时候,不要加biases的L2范数,会导致欠拟合)
# var.name: "InceptionV4/Logits/Logits/weights:0"
# var.op.name: "InceptionV4/Logits/Logits/weights"
# var.name.split(':')[0].split('/')[-1]
for var in tf.trainable_variables():
#if (var.name.split(':')[0].split('/')[-1] == "weights"):
if (var.op.name.split('/')[-1] == "weights"):
print (var)
print (var.get_shape())
print (sess.run(tf.nn.l2_loss(var)))
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息