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

一、错误描述

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。

你可能感兴趣的:(pytorch,Centernet,pytorch)