Tensorflow(steps,batch_size和句点)

时间:2018-03-20 13:06:45

标签: tensorflow machine-learning training-data

在tensorflow中训练模型时我有点困惑。 train_model函数中的步骤和batch_size和句点是什么? train_model函数的代码为in the link to Learn from google AI

1 个答案:

答案 0 :(得分:0)

如方法评论中所述,steps的数量是批次的前向和后向传递的数量。 此批次的大小(batch_size)是用于此培训过程的训练样本数。您只需计算batch_size个样本,就可以估算整个数据集的损失,从而逼近整个数据集的梯度。这个基本术语在神经网络的许多入门课程中都有解释。

此处period变量上的循环仅用于偶尔显示一些损失指标。