我已经从头开始为我的数据训练FCN32语义分段,我得到了以下输出:
可以看出,这不是一个很好的学习曲线,表明对数据的训练不当。
solver
如下:
net: "train_val.prototxt"
#test_net: "val.prototxt"
test_iter: 5105 #736
# make test net, but don't invoke it from the solver itself
test_interval: 1000000 #20000
display: 50
average_loss: 50
lr_policy: "step" #"fixed"
stepsize: 50000 #+
gamma: 0.1 #+
# lr for unnormalized softmax
base_lr: 1e-10
# high momentum
momentum: 0.99
# no gradient accumulation
iter_size: 1
max_iter: 600000
weight_decay: 0.0005
snapshot: 30000
snapshot_prefix: "snapshot/FCN32s_CNN1"
test_initialization: false
solver_mode: GPU
答案 0 :(得分:0)
您可以尝试改变学习率。良好的值通常介于0.1和0.0001之间。