上传者: 38674992
|
上传时间: 2023-02-28 19:33:33
|
文件大小: 42KB
|
文件类型: PDF
原文地址
分类目录——Pytorch
关键词:
Data.DataLoader、num_workers、batch
Pytorch在开进行小批量数据下降时开多进程报错
报错内容
RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly
去网上搜教程,说是在loader定义的时候多进程的参数指定的问题,我的loader定义如下
loader = Data.DataLoader(
dataset=torch_dataset, # torch TensorDataset format
ba