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