我在使用Tensorflow训练模型时有些困惑。train_model函数中的步骤(steps)、批量大小(batch_size)和周期(periods)是什么意思?train_model函数的代码可以在谷歌AI学习的链接中找到。
回答:
如方法注释中所述,steps
的数量是指对一个批次进行的前向和后向传递的次数。这个批次的大小(batch_size
)是用于此训练传递的训练样本数量。你通过仅在batch_size
个样本上计算来近似整个数据集的损失,因此也近似了梯度。这些基本术语在许多神经网络入门课程中都有解释。
这里对period
变量的循环只是用来偶尔显示一些损失指标。