3月 2019

tensorflow学习(3)

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