実行環境
- Windows10 Home
- torch 1.1.0
PytorchでDataLoaderを使おうとしたところ以下のエラーが発生
実行したプログラム
torch.utils.data.DataLoader(
dataset = dataset,
batch_size = 100,
shuffle = True,
num_workers = 8)
エラー文
RuntimeError: DataLoader worker (pid(s) ~) exited unexpectedly
原因
メモリ不足が原因
解決策
num_workersをエラーが出ない値まで小さくする
ex) num_workers = 0
等