解决方案
1.在主代码中加入如下代码
import torch
torch.multiprocessing.set_sharing_strategy('file_system')
2.设置num_workers,我原来的代码是
num_workers=multiprocessing.cpu_count()
但是这样子的话会导致CPU 资源过度使用,所以最好输出一下这个值然后减半什么的
解决方案
1.在主代码中加入如下代码
import torch
torch.multiprocessing.set_sharing_strategy('file_system')
2.设置num_workers,我原来的代码是
num_workers=multiprocessing.cpu_count()
但是这样子的话会导致CPU 资源过度使用,所以最好输出一下这个值然后减半什么的