我尝试使用自己的自定义数据来实现FCN-8s。从头开始训练时,我看到我的损失= -nan。
有人可以建议出现问题,我该如何纠正?我的solver.prototxt如下: -
train_val.prototxt与link above中给出的相同。我的自定义图像尺寸为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