データの前処理エラーpytorch

 

エラー：

トレースバック（最新の呼び出しの最後）：
  ファイル" train.py "、ライン305、で、<モジュール> 
    train_model（model_conv、基準、optimizer_conv、exp_lr_scheduler）
  ファイル" train.py "、ライン145、中train_model
     ための入力、age_labels、gender_labels 中dataloaders [相]：
  ファイルの" /home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py "、ライン637、中__next__
     リターンself._process_next_batch（バッチ）
  ファイル" /home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py "、ライン658、中_process_next_batch 
    レイズbatch.exc_type（batch.exc_msg）
RuntimeError：トレースバック（最新の呼び出しの最後）：
  ファイル" /home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py "、ライン138、中_worker_loopの
    サンプル = collat​​e_fn（[データセット[i]のための私でbatch_indices]）
  ファイル」/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py "行232、中default_collat​​e
     リターン [default_collat​​e（サンプル）のためのサンプルで転置] 
  ファイル" /ホーム/ホーム/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py "行232、中 <listcomp>
     リターン [default_collat​​e（サンプル）のためのサンプルで転置] 
  ファイル" /ホーム/ホーム/ anaconda3 /lib/python3.6/site-packages/torch/utils/data/dataloader.py "、ライン209は、中default_collat​​e
     リターン torch.stack（バッチは、0、アウト = アウト）
のRuntimeError：無効な引数0：以外と一致しなければなりませんテンソルのサイズで大きさ0。ガット224及び228  で寸法3 /pytorch/aten/src/TH/generic/THTensorMoreMath.cppで：1307

彼らは、入力のサイズと一致していないので、これは、関連するデータセットではなく、機能を前処理したデータと、次のとおりです。

transforms.Resize（input_size）

結果は、溶液の代わりに使用することである（224、224）外にないので、この機能は、おそらくデータセットの異なる解像度、スケーリングされます。

transforms.Resize（（input_size、input_size））

へ