2

VGG_ILSVRC_19_layers を数字の事前トレーニング済みモデルとして使用したいが、データセットは異なる。別のラベル ファイルが必要ですか? このモデルをアップロードしてデータセットに使用するにはどうすればよいですか?

私が得たVGG 16層のために

エラー: パラメータ 0 のウェイトをレイヤ 'fc6' からコピーできません。形の不一致。ソース パラメータの形状は 1 1 4096 25088 (102760448) です。ターゲット パラメータの形状は 4096 32768 (134217728) です。保存されたネットからコピーするのではなく、このレイヤーのパラメーターを最初から学習するには、レイヤーの名前を変更します。

レイヤーを変更する方法

4

1 に答える 1

1

ラベルは、モデルではなく、DIGITS のデータセットに関連付けられています。事前トレーニング済みの VGG モデルをアップロードする場合、モデルが 1000 ではなく N クラスで機能するように、おそらく最後の内積レイヤーの名前を変更する必要があります (この回答を参照)。

于 2016-10-19T16:55:32.437 に答える