训练VGG-16需要多少纪元

时间:2017-05-01 14:31:59

标签: parameters tensorflow neural-network deep-learning image-recognition

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

训练率 - 0.01,

重量衰减 - 0.0005,

动量 - 0.9,

批量大小 - 64,

我已经训练了大约三天。但是训练的准确性在20个时期后保持不变,约为15%-20%。任何人都可以给我一些提示来提高准确性吗?

1 个答案:

答案 0 :(得分:8)

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

之后

训练率 - 0.001,

重量衰减 - 0,

动量 - 0.9,

批量大小 - 64。

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