【Centernet复现】AttributeError:Can‘t pickle local object ‘get_dataset.＜locals＞.Dataset

榴莲味的电池

已于 2025-05-24 21:38:50 修改

阅读量2.3k

点赞数 13

分类专栏： pytorch 文章标签： Centernet pytorch

于 2021-04-28 17:02:15 首次发布

本文链接：https://blog.youkuaiyun.com/qq_43115981/article/details/116237860

版权

pytorch 专栏收录该内容

8 篇文章

订阅专栏

一、错误描述

AttributeError:Can't pickle local object 'get_dataset.<locals>.Dataset

二、解决办法
把涉及到numworks的值都改为0。
opts.py文件里的参数num_workers默认值改为0。
main.py文件里num_workers的值改为0。

  val_loader = torch.utils.data.DataLoader(
      Dataset(opt, 'val'), 
      batch_size=1, 
      shuffle=False,
      num_workers=0,
      pin_memory=True
  )

其他的自己检查一下，把涉及到num_workers的值都改为0。