1. ホーム
  2. ピトーチ

[Centernet recurrence] AttributeError:Can't pickle local object 'get_dataset.<locals>.Dataset

2022-02-10 17:28:05
<パス

I. エラーの説明

AttributeError:Can't pickle local object 'get_dataset.<locals>.Dataset


II. 解決方法
に関係するコードを numworks の値は 0 に変更されます。
opts.pyファイル内のパラメータ num_workers デフォルト値は0に変更されます。
main.py ファイルで num_workers が0に変更されています。

  val_loader = torch.utils.data.DataLoader(
      Dataset(opt, 'val'), 
      batch_size=1, 
      shuffle=False,
      num_workers=0,
      pin_memory=True
  )


あとは自分で確認し、以下のコードに関連するものを入れてください。 num_workers の値を0にします。