本文介绍了如何在TensorFlow中实现模型的按需加载，即在创建Saver时指定仅加载部分参数，从而提高效率或针对特定任务优化。
在create saver之前加入下面的代码
t_variables = tf.trainable_variables()
print("t_variables", t_variables)
# var_to_restore = [v for v in t_variables if not v.name.startswith('InceptionResnetV1/Bottleneck')]
var_to_restore2 = [v for v in t_variables if not v.name.st
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
