Pytorch で Iterable.next() すると停止する

原因: num_workersの設定が正しくない

対策:num_workers=0 にする

こんな感じ

dataloader = DataLoader(trainset, batch_size=5, shuffle=True, num_workers=0)

 

以下:公式ドキュメントより引用

num_workers (python:intoptional) – how many subprocesses to use for data loading. 0 means that the data will be loaded in the main process. (default: 0)

つまり、下手にサブプロセスで実行しようとしてミスると停止する