制作批量张量流

时间:2017-10-24 04:20:14

标签: python tensorflow batch-processing

所以我遇到了在我的代码中进行批处理的问题,问题是,我试图搜索我们如何进行批处理,但我发现在MNIST示例程序中使用了一些像next_batch这样的方法。我真的很感激,如果有人真的可以给我一些关于如何在我的程序中批量生产的提示。

{{1}}

当然,这个项目的结果是不对的。

1 个答案:

答案 0 :(得分:0)

继续提取您的数据批次并继续将其提供给网络进行培训。在每个纪元中,训练数据集的所有样本都应该运行一次。所以你可以像这样重写你的代码:

仅限代码部分:

epochs = 4000
batch_size = 100
for epoch_no in range(epochs):
    for index, offset in enumerate(range(0, 86594, batch_size)):
        xs, ys = train_x[offset: offset + batch_size], train_t[offset: offset + batch_size]
        sess.run(train_step, feed_dict={x: xs, t: ys})

        if index % 100 == 0:
            loss_val, acc_val = sess.run([loss, accuracy], feed_dict = {x: test_x, t: test_t})
            print ('Epoch %d, Step: %d, Loss: %f, Accuracy: %f'% (epoch_no, index, loss_val, acc_val))