【发布时间】:2016-07-31 07:58:19
【问题描述】:
我正在尝试训练一个非常大的模型。因此,我只能将非常小的批量大小放入 GPU 内存中。处理小批量的结果非常noisy gradient estimations。
我该怎么做才能避免这个问题?
【问题讨论】:
标签: machine-learning neural-network deep-learning caffe gradient-descent
我正在尝试训练一个非常大的模型。因此,我只能将非常小的批量大小放入 GPU 内存中。处理小批量的结果非常noisy gradient estimations。
我该怎么做才能避免这个问题?
【问题讨论】:
标签: machine-learning neural-network deep-learning caffe gradient-descent
您可以更改求解器参数中的iter_size。
Caffe 在每个随机梯度下降步骤中累积 iter_size x batch_size 实例上的梯度。
所以在内存有限无法使用大batch_size的情况下,增加iter_size也可以获得更稳定的梯度。
【讨论】:
正如this post 中所述,批量大小在理论上不是问题(随机梯度下降的效率已在批量大小为 1 的情况下得到证明)。确保您正确实施批次(样本应在您的数据中随机选取)。
【讨论】: