1

我正在使用 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,依此类推)

迭代 100

迭代 500

这是我第一次训练 ResNet。如果您有任何建议,请告诉我。非常感谢!

4

0 回答 0