微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

无法在 Pytorch 数据集中打开共享文件映射错误

如何解决无法在 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 举报,一经查实,本站将立刻删除。