学习动手深度学习时跑代码出现标题问题
查了下问题出于进程上。
出错代码:
batch_size = 256
def get_dataloader_workers():
"""使用4个进程来读取数据。"""
return 4
train_iter = data.DataLoader(mnist_train, batch_size, shuffle=True,
num_workers=get_dataloader_workers())#num_workers给多少进程
timer = d2l.Timer()
for X, y in train_iter:
continue
f'{timer.stop():.2f} sec'
在windows中是不能使用多个子进程加载数据的,在linux系统中可以。所以在windows中要将num_workers设置为0的。
解决方案:将return 4 改成return 0。