Caffe框架:batch size的真实例子,max_iter, test_iter, epoch

Caffe framework: A real example of batch size, max_iter, test_iter, epoch

设置如下:

test_observations : 6,767;
train_observations: 73,268;
train/test batch_size = 50;

我该如何设置batch_sizetest_itertest_intervalmax_iter

谢谢!

因此您的验证大小为 6,767,您的验证批量大小为 50。 你的 test_iter = validation set/ validation_batch_size = 6,767/50 = 135 (approx.) 所以它几乎覆盖了验证集。和测试间隔,您可以选择任何值 - 它是您的网络将在验证集上测试性能的迭代次数。对于更大的网络,test_interval 使用 5k 这样的值。对于您的网络 test_interval of 1000 似乎没问题。

为了找到 max_iter,你必须选择你想要的 epoch 数,即你想要覆盖你的训练规模的次数(假设为 2 - 明智地选择这个数字而不是过拟合网络)。还有一件事目前在caffe中没有epoch的实现,但是从这个公式可以看出它的效果。 max_iter = #epochs * (训练 set/training_batch_size) = 2 * (73,268/50) = 29,000(大约)。这样它就会遍历你的训练集两次,在训练完 1k 张图像后,它会在你的 6,767 张图像上进行验证以进行优化。