azhao_Y 2020-03-16 09:50 采纳率: 0%
浏览 4635

TensorFlow训练过程中几步训练损失和准确率就一直不变了

图片说明

从这步开始训练就一直不变了,很奇怪

train部分代码:
```# 导入文件
import os
import numpy as np
import tensorflow as tf
import input_data
import model

变量声明

N_CLASSES = 4 # 四种花类型
IMG_W = 64 # resize图像,太大的话训练时间久
IMG_H = 64
BATCH_SIZE = 20
CAPACITY = 200
MAX_STEP = 10000 # 一般大于10K
learning_rate = 0.00005 # 一般小于0.0001

获取批次batch

train_dir = 'D:/python/four_flower-master 5/input_data' # 训练样本的读入路径
logs_train_dir = 'D:/python/four_flower-master 5/save' # logs存储路径

train, train_label = input_data.get_files(train_dir)

train, train_label, val, val_label = input_data.get_files(train_dir, 0.3)

训练数据及标签

train_batch, train_label_batch = input_data.get_batch(train, train_label, IMG_W, IMG_H, BATCH_SIZE, CAPACITY)

测试数据及标签

val_batch, val_label_batch = input_data.get_batch(val, val_label, IMG_W, IMG_H, BATCH_SIZE, CAPACITY)

训练操作定义

train_logits = model.inference(train_batch, BATCH_SIZE, N_CLASSES)
train_loss = model.losses(train_logits, train_label_batch)
train_op = model.trainning(train_loss, learning_rate)
train_acc = model.evaluation(train_logits, train_label_batch)

测试操作定义

test_logits = model.inference(val_batch, BATCH_SIZE, N_CLASSES)
test_loss = model.losses(test_logits, val_label_batch)
test_acc = model.evaluation(test_logits, val_label_batch)

这个是log汇总记录

summary_op = tf.summary.merge_all()

产生一个会话

sess = tf.Session()

产生一个writer来写log文件

train_writer = tf.summary.FileWriter(logs_train_dir, sess.graph)

val_writer = tf.summary.FileWriter(logs_test_dir, sess.graph)

产生一个saver来存储训练好的模型

saver = tf.train.Saver()

所有节点初始化

sess.run(tf.global_variables_initializer())

队列监控

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess=sess, coord=coord)

进行batch的训练

try:
# 执行MAX_STEP步的训练,一步一个batch
for step in np.arange(MAX_STEP):
if coord.should_stop():
break
_, tra_loss, tra_acc = sess.run([train_op, train_loss, train_acc])

    # 每隔50步打印一次当前的loss以及acc,同时记录log,写入writer
    if step % 10 == 0:
        print('Step %d, train loss = %.2f, train accuracy = %.2f%%' % (step, tra_loss, tra_acc * 100.0))
        summary_str = sess.run(summary_op)
        train_writer.add_summary(summary_str, step)
    # 每隔100步,保存一次训练好的模型
    if (step + 1) == MAX_STEP:
        checkpoint_path = os.path.join(logs_train_dir, 'model.ckpt')
        saver.save(sess, checkpoint_path, global_step=step)

except tf.errors.OutOfRangeError:
print('Done training -- epoch limit reached')

finally:
coord.request_stop()




  • 写回答

3条回答 默认 最新

  • threenewbee 2020-03-16 11:36
    关注

    你正确率都100%了还想怎么变?

    评论

报告相同问题?

悬赏问题

  • ¥15 基于卷积神经网络的声纹识别
  • ¥15 Python中的request,如何使用ssr节点,通过代理requests网页。本人在泰国,需要用大陆ip才能玩网页游戏,合法合规。
  • ¥100 为什么这个恒流源电路不能恒流?
  • ¥15 有偿求跨组件数据流路径图
  • ¥15 写一个方法checkPerson,入参实体类Person,出参布尔值
  • ¥15 我想咨询一下路面纹理三维点云数据处理的一些问题,上传的坐标文件里是怎么对无序点进行编号的,以及xy坐标在处理的时候是进行整体模型分片处理的吗
  • ¥15 CSAPPattacklab
  • ¥15 一直显示正在等待HID—ISP
  • ¥15 Python turtle 画图
  • ¥15 stm32开发clion时遇到的编译问题