原因: num_workersの設定が正しくない
対策:num_workers=0 にする
こんな感じ
dataloader = DataLoader(trainset, batch_size=5, shuffle=True, num_workers=0)
以下:公式ドキュメントより引用
num_workers (python:int, optional) – how many subprocesses to use for data loading.
0
means that the data will be loaded in the main process. (default: 0
)
つまり、下手にサブプロセスで実行しようとしてミスると停止する