我正在使用包含 3k 图像的数据集从头开始训练 VGG-16 模型。
我使用 Tensorflow 平台和 8 个 CPU,没有任何 GPU。
训练率 - 0.01,
重量衰减 - 0.0005,
动量 - 0.9,
批量大小 - 64,
我已经训练了大约三天。但是训练准确率一直没有改变,在 20 个 epoch 后大约在 15%-20%。谁能给我一些提示以提高准确性?
最佳答案
似乎我使用了太大的学习率。或者重量衰减不像它 promise 的那样起作用。在我将我的超参数更改为后,
训练率 - 0.001,
重量衰减 - 0,
动量 - 0.9,
批量大小 - 64。
一切都很好。现在我可以在 25 个 epoch 后获得大约 90% 的准确率。仅供引用。
关于parameters - 训练 VGG-16 需要多少个 epoch,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43720817/