我正在尝试使用我自己的自定义数据来实现FCN-8 。在训练时,从头开始,我看到我的损失 = -nan。
有人可以建议出了什么问题,我该如何纠正?我的solver.prototxt如下:-
train_val.prototxt 与上面链接中给出的相同。我的自定义图像大小为 3x512x640,标签为 1x512x640。标签有 11 种不同类型。
net: "/home/ubuntu/CNN/train_val.prototxt"
test_iter: 13
test_interval: 500
display: 20
average_loss: 20
lr_policy: "fixed"
base_lr: 1e-4
momentum: 0.99
iter_size: 1
max_iter: 3000
weight_decay: 0.0005
snapshot: 200
snapshot_prefix: "train"
test_initialization: false