分类：num_workers参数

2021
07-21

pytorch DataLoader的num_workers参数与设置大小详解

Q：在给Dataloader设置worker数量（num_worker）时，到底设置多少合适？这个worker到底怎么工作的？train_loader=torch.utils.data.DataLoader(train_dataset,batch_size=batch_size,shuffle=True,num_workers=4)参数详解：1、每次dataloader加载数据时：dataloader一次性创建num_worker个worker，（也可以说dataloader一... 继续阅读 >