Caffe训练迭代损失是-nan

时间:2015-11-18 12:22:40

标签: deep-learning caffe conv-neural-network

我尝试使用自己的自定义数据来实现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

0 个答案:

没有答案