tensorflow学习(3)

参考链接。进行训练时,如果需要选择指定的变量权重更新,可以使用:vars=tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES,"re")获取所有包含"re"的变量,然后使用optimizer.minimize(cost,vars)更新权重。

发表评论

电子邮件地址不会被公开。 必填项已用*标注