如何解决无法在 Pytorch 数据集中打开共享文件映射错误
在 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 举报,一经查实,本站将立刻删除。