TensorFlow参数:epoch,iteration,batchsize

深度学习TensorFlow 579

深度学习中经常看到epochiterationbatchsize,下面按自己的理解说说这三个的区别:

  • batchsize:批大小。在深度学习中,一般采用SGD训练,即每次训练在训练集中取batchsize个样本训练;
  • iteration:1个iteration等于使用batchsize个样本训练一次;
  • epoch:1个epoch等于使用训练集中的全部样本训练一次;

举个例子,训练集有1000个样本,batchsize=10,那么:

训练完整个样本集需要:

100次iteration,1次epoch。

出处:https://blog.csdn.net/sinat_30071459/article/details/50721565

扩展阅读:

Post Comment