0

pytorch を使用して resnet50 モデルをトレーニングしているときに、次のエラーが発生しました。

RuntimeError: 無効な引数 0: テンソルのサイズは次元 0 を除いて一致する必要があります。/pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333 で次元 1 で 3 と 1 を取得しました

私はこれを使用しています: http://github.com/Helias/Car-Model-Recognition/

このデータセットでhttp://vmmrdb.cecsresearch.org/

4

1 に答える 1