pytorch错误解决: BrokenPipeError: [Errno 32] Broken pipe
生活随笔
收集整理的這篇文章主要介紹了
pytorch错误解决: BrokenPipeError: [Errno 32] Broken pipe
小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
該問(wèn)題的產(chǎn)生是由于windows下多線程的問(wèn)題,和DataLoader類有關(guān)
原因是windows 下 DataLoader加載數(shù)據(jù)集啟用線程的數(shù)目這一塊,是有bug的
trainloader = DataLoader(dataset_train, batch_size=batch_size,shuffle=True, num_workers=1) testloader = DataLoader(dataset_test, batch_size=batch_size,shuffle=False, num_workers=1)解決方法很簡(jiǎn)答,就是把原代碼中的num_workers=1改為=0即可
總結(jié)
以上是生活随笔為你收集整理的pytorch错误解决: BrokenPipeError: [Errno 32] Broken pipe的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 数学知识复习:二阶导复合函数的链式法则
- 下一篇: pytorch笔记:torch.nn.G