问题描述
在 Pytorch 1.4 中使用自定义数据集训练模型时,在看似随机的 epoch 数之后抛出以下错误。
RuntimeError: Couldn't open shared file mapping: <torch_15324_2327643205>,error code: <1455>
数据集被包裹在一个 torch.utils.data.DataLoader
中,并使用 4 个工人,相当于物理内核的数量。
class TSNDataSet(data.Dataset):
def __init__(self,pickle_file_paths,transforms):
self.pickle_file_paths = pickle_file_paths # list with file paths to pickle files
self.dataset_size = len(pickle_file_paths)
def __getitem__(self,index):
with open(self.pickle_file_paths[index],'rb') as f:
mffs = pickle.load(f)
return mffs,index
def __len__(self):
return self.dataset_size
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)