問題タブ [image-classification]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
392 問題
0
投票する
1
に答える
182
参照
pytorch - 入力 batch_size (18) がターゲット batch_size (6) と一致すると予想される
画像分類用の RNN はグレー画像でのみ使用できますか? 次のプログラムは、グレー イメージの分類に使用できます。
RGB 画像を使用すると、次のエラーが発生します。
入力 batch_size (18) がターゲット batch_size (6) と一致すると予想される
この行でloss = criterion(outputs, labels)。
train、valid、および test のデータ読み込みは次のとおりです。
私のモデルは次のとおりです。
印刷イメージと出力形状は
間違いはどこですか？