在tensorflow中训练模型时我有点困惑。 train_model函数中的步骤和batch_size和句点是什么? train_model函数的代码为in the link to Learn from google AI
答案 0 :(得分:0)
如方法评论中所述,steps
的数量是批次的前向和后向传递的数量。
此批次的大小(batch_size
)是用于此培训过程的训练样本数。您只需计算batch_size
个样本,就可以估算整个数据集的损失,从而逼近整个数据集的梯度。这个基本术语在神经网络的许多入门课程中都有解释。
此处period
变量上的循环仅用于偶尔显示一些损失指标。