独自のカスタム データを使用して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