Caffe框架:批量大小,max_iter,test_iter,epoch的真实示例

时间:2016-09-17 07:12:39

标签: machine-learning neural-network deep-learning caffe

以下是设置:

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

我应该如何设置batch_sizetest_itertest_intervalmax_iter

谢谢!

1 个答案:

答案 0 :(得分:1)

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

要查找max_iter,您必须选择您想要去的时期数,即您希望覆盖训练大小的次数(假设为2) - 明智地选择此数字不要过度适应网络。还有一件事,目前没有在caffe中实现epoch,但是它的效果可以从这个公式中看出来。 max_iter = #epochs *(训练集/ training_batch_size)= 2 *(73,268 / 50)= 29,000(大约)。这样它将会超过你的训练集两次,并且在训练1k图像后,它将验证你的6,767张图像以进行优化。