训练神经网络太慢 DataLoader设置num_workers=4报错 RuntimeError: Caught RuntimeError in DataLoader worker process
在服务器上跑网络感觉很慢,尤其是几块显卡都有人在跑的时候,尝试在train_loader=DataLoader(train_data,batch_size=trainargs.batch_size,shuffle=True,num_workers=4) 设置num_workers=4,发现报错RuntimeError: Caught RuntimeError in DataLoader worker process。于是网上搜答案,但却没找到能成功解决的。于是放弃,就让它这么慢吧。直到有一天,心有不..
原创
2021-07-04 10:37:38 ·
10711 阅读 ·
0 评论