使用TFRecord与Dataset的TensorFlow MNIST模型准确率低下问题咨询

我借助TensorFlow官方提供的脚本将MNIST数据集转换成了TFRecord格式，之后编写了以下代码来实现TFRecord文件的读取、神经网络模型的构建以及训练流程：

import tensorflow as tf

BATCH_SIZE = 32
epoch = 20
n_hidden_1 = 256  # 第一层神经元数量
n_hidden_2 = 256  # 第二层神经元数量
num_input = 784    # MNIST数据输入（图像尺寸：28*28）
num_classes = 10   # MNIST总类别数（0-9数字）
# 后续模型构建与训练代码...

内容的提问来源于stack exchange，提问作者Thien