2016-10-20 22 views
2

私はDGITSにプリトレインモデル(VGG 19)を使用しようとしましたが、このエラーが発生しました。配備ネットワークにSoftmaxレイヤーがありません

ERROR: Your deploy network is missing a Softmax layer! Read the documentation for custom networks and/or look at the standard networks for examples

私は2つのクラスしか持たないデータセットでテストしようとします。

私はthisthisを最後のレイヤーに変更しようとしましたが、エラーが発生しました。新しいデータセットに基づいてレイヤーを変更するにはどうすればよいですか?

は私が最後の層を変更しようと、私はエラー

ERROR: Layer 'softmax' references bottom 'fc8' at the TRAIN stage however this blob is not included at that stage. Please consider using an include directive to limit the scope of this layer.

答えて

2

本当にアップロードする必要があるときは、あなたが「電車/ヴァル」ネットワークをアップロードしようとしているため、問題を抱えているのです「オールインワン」ネットワーク残念ながら、私たちはこれをうまく文書化していません。私はan RFEを作成して、ドキュメントの改善を促しました。

このような何か見て、ネットワーク内の最後の層を調整してみてくださいhere私はオールインワンネットにカフェの例ネットワークの更新に提案しました、とhereがどのようにされ、詳細についてはhttps://github.com/NVIDIA/DIGITS/blob/v4.0.0/digits/standard-networks/caffe/lenet.prototxt#L162-L184

をどのように私はオールインワンネットにデフォルトのDIGITSネットワークを更新しました。

+0

ありがとうございますが、最後のレイヤーを変更したときにエラーが発生しました。 "ERROR:層 'softmax'はTRAIN段階で底 'fc8'を参照しますが、この段階ではこのBLOBは含まれていません。この層の範囲を制限するためにインクルードディレクティブを使用することを検討してください。最後のレイヤーはどのように変更できますか? –

+0

最後のレイヤーはどのように変更できますか? 「オールインワン」はどのように使用できますか?ファインチューニングモデルに関連する「オールインワン」ですか? VGG 19層を使用するために私はどのように2つのクラスだけを持つ私のデータセットで使用するために最後の層を変更できますか? –

+1

あなたは[この回答](http://stackoverflow.com/a/33773152/1714410)便利な:prototxtを「配備する」prototxtに変換する方法があります。 – Shai

関連する問題