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)

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

 

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

このサイトはスパムを低減するために Akismet を使っています。コメントデータの処理方法の詳細はこちらをご覧ください