num_workers

num_workers是使用多少个子进程来导入数据。设置为0,就是使用主进程来导入数据。注意:这个数字必须是大于等于0的,负数估计会出错

from torch.utils.data import DataLoader

train_loader = DataLoader(dataset=train_data, batch_size=train_bs, shuffle=True, num_worker=4)

valid_loader = DataLoader(dataset=valid_data, batch_size=valid_bs, num_worker=4)

num_workers通过影响数据加载速度,从而影响训练速度

在调整num_workers的过程中,发现训练速度并没有变化


版权声明:本文为baidu_38262850原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。