tensorflow 只恢复部分模型参数的实例
我就废话不多说了，直接上代码吧！
importtensorflowastf
defmodel_1():
withtf.variable_scope("var_a"):
a=tf.Variable(initial_value=[1,2,3],name="a")
vars=[varforvarintf.trainable_variables()ifvar.name.startswith("var_a")]
print(len(vars))
returnvars
defmodel_2():
vars1=model_1()
withtf.variable_scope("var_b"):
a=tf.Variable(initial_value=[1,2,3],name="a")
vars2=[varforvarintf.trainable_variables()ifvar.name.startswith("var")]
print(len(vars2))
returnvars1
defpretrain_model1():
print("--------model1------")
vars=model_1()
withtf.Session()assess:
sess.run(tf.global_variables_initializer())
saver=tf.train.Saver()
saver.save(sess,"./model.ckpt")
deftrain_model2():
print("--------model2------")
model1_vars=model_2()
withtf.Session()assess:
sess.run(tf.global_variables_initializer())
saver=tf.train.Saver(var_list=model1_vars)
saver.restore(sess,"./model.ckpt")
vars=sess.run([model1_vars])
forvarinvars:
print(var)
step=2
ifstep==1:
pretrain_model1()
else:
train_model2()
以上这篇tensorflow只恢复部分模型参数的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。
热门推荐
10 财运虎年祝福语简短语
11 单亲爸爸春节祝福语简短
12 狗狗上厕所训练
13 植物种植祝福语简短
14 翻译考试祝福语朋友简短
15 个人周年庆简短祝福语
16 狗狗怎么训练
17 分别祝福语简短搞笑英文
18 生日广州男生祝福语简短