tensorflow global_step不自增一直为1解决

训练的时候想要获取训练的步数,最后发现每次获得的global_step的值都是1,最后发现自己代码写错了,把变量初始化写在循环之中了,导致每次都要初始化步数,导致每次的步数都是1

if __name__=='__main__':
    #这里都是在定义 并没有初始化
    train_op, cross_entropy, accuracy, global_step, summary_op, init_op, y ,x,y_,lr= get_net()
    #从里面获得数据
    with tf.Session() as sess:
        ##########################################################
        ####sess.run([init_op, tf.local_variables_initializer()]) 初始化要放在最前面,自己放在了循环里面 不断初始化 导致step不变
        ##########################################################
        sess.run([init_op, tf.local_variables_initializer()])  # , tf.local_variables_initializer()])
        ################### dataset的方法
        i =0
        while True:
            try:
                # sess.run(tf.initialize_all_variables()) 自己原来的初始化放在这里
                #为什么global_step不对
                _, loss_value,step,input,label,lear= sess.run([train_op, cross_entropy,global_step,x,y_,lr])
                print(loss_value,step,input.shape,label.shape,lear)
            except tf.errors.OutOfRangeError:
                break

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值