我正在使用 Caffe 的 ResNet-50 对图像进行分类。我可以使用 ResNet-50_model.caffemodel 对其进行微调,我设置的初始学习率为 0.001,但每 500 次迭代后的学习率除以 10 而不是减少 0.00005。
这是我的 resnet_solver.prototxt
net: "data/resnet_data/resnet_50.prototxt"
test_iter: 320
test_interval: 100
base_lr: 0.001
lr_policy: "step"
gamma: 0.1
stepsize: 500
display: 100
max_iter: 2000
momentum: 0.9
weight_decay: 0.0005
snapshot: 500
snapshot_prefix: "data/resnet_data/model/resnet_grid"
solver_mode: GPU
这是训练截图:(iter 100 lr = 0.001,iter 500 lr = 0.0001,iter 1000 lr = 1e-05,依此类推)
这是我第一次训练 ResNet。如果您有任何建议,请告诉我。非常感谢!