解决pytorch DataLoader num_workers出现的问题
今天小编就为大家分享一篇解决pytorch DataLoader num_workers出现的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
Pytorch Dataloader的num_workers设置建议
Dataloader参数设置示例:Dataloader(dataset,num_workers=8,pin_memory=True)Dataloader的num_worker设置多少才合适,这个问题是很难有一个推荐的值。有以下几个建议:num_workers=0表示只有主进程去加载batch数据,这...