在caffe框架中模型训练之前,需要配置solver.prototxt文件做一些配置,现对其做一些讲解:
1.test_iter
test_iter=测试集大小/测试层中batch_size
2.max_iter
max_iter=test_iter*80,其中80是经验值,你可以调大或调小一些,总之使在max_iter次迭代训练中,模型要收敛即可
3.test_interval
test_interval=训练集大小/训练层batch_size大小
4.base_lr
一般保持原始值不变,也可以适当调整
5.solver_mode
若是用CPU训练则写为CPU,若是用GPU跑模型,则写为GPU;GPU比CPU快很多,难以想象
Caffe模型训练之solver.prototxt配置
最新推荐文章于 2019-12-11 17:19:36 发布