tensorflow保存和重载模型

1.保存一个模型

注意要给变量命名,要把op用tf.add_to_collection()添加
saver = tf.train.Saver(max_to_keep=20)  #注意要在初始化变量以后,迭代以前,参数为最大保存数量
saver.save(sess, 'my_test_model',global_step)#在迭代中保存模型
第一个参数是保存模型,第二个参数是保存路径,第三个参数是保存多次模型时用来命名
    注意:要把导入模型时需要用到的(例如需要feed_dict的变量,以及op命名(name=''))
2.导入一个训练好的模型

new_saver = tf.train.import_meta_graph('my_test_model-1000.meta')#在创建会话以后
new_saver.restore(sess,'../model/model_LR_test')#加载模型中各种变量的值,注意这里不用文件的后缀
op=tf.get_collection('op_name')[0]
graph = tf.get_default_graph() 
X = graph.get_operation_by_name('X').outputs[0]#为了将placeholder加载出来

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值