pytorch

最新文章

Pytorch Dataloader的num_workers设置建议

Dataloader 参数设置示例:Dataloader(dataset, num_workers=8, pin_memory=True)Dataloader的num_worker设置多少才合适,这个问题是很难有一个推荐的值。有以下几个建议:num_workers=0表示只有主进程去加载batch数据,这个可能会是一个瓶颈。num_workers = 1表示只有一个worker进程用来加载batc