tensorflow在save model的时候,如果选择了global_step选项,会把对应的学习率也保存下来,然后restore的时候会把学习率也恢复,此时可以在restore的时候指定恢复的tensor名字就行,通过saver.restore(var_list=var_list)就行,这样就忽略了学习率,仅恢复了权重. 但需要注意的是, 在restore的时候,如果是通过tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)得到的var_list, 并没有指定变量的scope, 并且global_step trainable属性并没有设置为False, 那么在通过saver.restore(var_list = var_list)时,会将当时存下的模型时的global_step和learning_rate值同样恢复.
tensorflow save使用
最新推荐文章于 2025-12-30 11:38:06 发布
本文详细介绍了在使用TensorFlow进行模型训练时,如何利用global_step选项保存模型,并在恢复模型时选择性地忽略学习率,仅恢复权重。特别强调了在restore过程中通过指定var_list来控制恢复哪些变量的重要性。
2万+

被折叠的 条评论
为什么被折叠?



