首页 文章

训练VGG-16需要多少个时代

提问于
浏览
0

我正在使用包含3k图像的数据集从头开始训练VGG-16模型 . 我使用Tensorflow平台和8 cpu没有任何gpu .

训练率 - 0.01,

重量衰减 - 0.0005,

动量 - 0.9,

批量大小 - 64,

我已经训练了大约三天 . 但是训练的准确性在20个时期后保持不变,约为15%-20% . 谁能给我一些提示来提高准确性?

1 回答

  • 6

    好像我使用了太大的学习率 . 或者体重衰减不会像它承诺的那样起作用 . 我将超级参数更改为

    训练率 - 0.001,

    重量衰减 - 0,

    动量 - 0.9,

    批量大小 - 64 .

    Eveything很好 . 现在,我可以在25个时代后获得90%的准确率 . 仅供参考 .

相关问题