训练 VGG-16 需要多少个 epoch
How many epochs does it take to train VGG-16
我正在使用包含 3k 图像的数据集从头开始训练 VGG-16 模型。
我使用 Tensorflow 平台和 8 个 cpu,没有任何 gpu。
训练率 - 0.01,
权重衰减 - 0.0005,
动量 - 0.9,
批量大小 - 64,
我坚持训练了三天左右。但训练准确率一直没有变化,在 20 个 epoch 后保持在 15%-20% 左右。谁能给我一些提高准确性的提示?
看来我用的学习率太大了。或者重量衰减并没有像它承诺的那样起作用。在我将超参数更改为
训练率 - 0.001,
权重衰减 - 0,
动量 - 0.9,
批量大小 - 64。
一切都很好。现在我可以在 25 个 epoch 后获得大约 90% 的准确率。仅供参考。
我正在使用包含 3k 图像的数据集从头开始训练 VGG-16 模型。 我使用 Tensorflow 平台和 8 个 cpu,没有任何 gpu。
训练率 - 0.01,
权重衰减 - 0.0005,
动量 - 0.9,
批量大小 - 64,
我坚持训练了三天左右。但训练准确率一直没有变化,在 20 个 epoch 后保持在 15%-20% 左右。谁能给我一些提高准确性的提示?
看来我用的学习率太大了。或者重量衰减并没有像它承诺的那样起作用。在我将超参数更改为
训练率 - 0.001,
权重衰减 - 0,
动量 - 0.9,
批量大小 - 64。
一切都很好。现在我可以在 25 个 epoch 后获得大约 90% 的准确率。仅供参考。