TensorFlow保存训练模式

5 篇文章 0 订阅
3 篇文章 0 订阅

TensorFlow训练模式如何保存,网上很多列子,拿来一试竟然没有作用! 后面自己摸索了下,搞定了,下面列子每次启动训练都会读取上次训练结果,每次训练也会自动存储。做一个记录吧,方便日后查询:

W_fc2 = weight_variable('W_fc2',[1024,64])  
b_fc2 = bias_variable([64]) 

W_fc3 = weight_variable('W_fc3',[64, 6])  
b_fc3 = bias_variable([6]) 

F1=tf.matmul(h_fc1_drop, W_fc2) + b_fc2
F2=tf.matmul(F1, W_fc3) + b_fc3
y_conv=tf.nn.softmax(tt) 

#下面代码检查创建保存文件路径
chk_dir="D:\\PokerChkPointCNN"
if(not os.path.exists(chk_dir)):
    os.makedirs(chk_dir)
saver=tf.train.Saver()

tf.global_variables_initializer().run()

#主意一定要在前面代码执行后才执行下面代码,否则是无意义的,也就是无法恢复上次的训练结果
ckpt=tf.train.latest_checkpoint(chk_dir)
if(ckpt):
  saver.restore(sess,ckpt)

while true:
    #训练过程中的保存
    saver.save(sess, chk_dir+"\\model.ckpt",global_step=i+1)  

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值