matlab - 为什么 vgg19 的训练速度比 alexnet 慢得多，分别是 50 小时和 4 小时？

Question

我正在尝试使用预训练模型vgg19并alexnet作为特征提取器，然后svm用于分类。我有大约 15000 张图像作为输入数据。我正在使用服务器 hp proliant g7，但vgg19需要大约 50 小时来训练，而alexnet只需要大约 4 小时。这是正常的，还是我应该针对vgg19此案调查一些设置或数据问题？

标签： matlabsvmfeature-extractionpre-trained-modelvgg-net

Answer 1

Vgg19 是比 alexnet 大得多的网络，如果我没记错的话，大约 12 倍。我怀疑 alexnet 可以毫无问题地融入内存，但 vgg19 却不行。如果 MATLAB 在训练期间内存不足，它可能会开始与硬盘交换内存，这会变得非常缓慢（而不是像 Tensorflow 那样立即因内存不足消息而崩溃）。我建议减少小批量大小或尝试其他网络。