tensorflow retrain.py了解train_batch_size

时间:2018-03-07 06:29:43

标签: tensorflow

我正在通过Tensorflow InceptionV3教程:https://www.tensorflow.org/tutorials/image_retraining#bottlenecks

我遇到以下段落:

  

默认情况下,此脚本将运行4,000个训练步骤。每个步骤从训练集中随机选择十个图像,从缓存中找到它们的瓶颈,并将它们输入到最后一层以获得预测。然后将这些预测与实际标签进行比较,以通过反向传播过程更新最终图层的权重。

随机拍摄"十张图像"是train_batch_size=10?同时在源代码中我发现了这个:

parser.add_argument(
    '--train_batch_size',
    type=int,
    default=100,
    help='How many images to train on at a time.'
)

这是否意味着我不正确地解释段落?如果是这样,train_batch_size意味着什么,它与十个随机图像有什么不同?或者它只是意味着教程页面与实际代码过时了?

源代码:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py

1 个答案:

答案 0 :(得分:1)

原来这是一个错字。实际上,10个随机图像应该是100个随机图像,对应于train_batch_size

提出解决问题的请求: https://github.com/tensorflow/tensorflow/pull/17638