Skip to content

ch1.4训练一个分类器中可能存在多进程报错的问题 #145

Open
@hebf181

Description

@hebf181

使用的是macos系统。在运行过程中存在多进程报错的问题,应该在
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
shuffle=True, num_workers=0)
以及

testloader = torch.utils.data.DataLoader(testset, batch_size=4,
shuffle=False, num_workers=0)
中设置进程数目为0
或者在训练的过程中使用if name=='main':将训练过程包起来。这样就可以正常运行不报错了。

多进程加载数据在windows系统中也不支持,在linux中可以。
转载自CSDN。
pytorch小白,欢迎交流

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions