加载预训练模型时KeyError: ‘model‘

加载预训练模型时KeyError: ‘model‘_第1张图片

self.load_state_dict(torch.load(pretrained, map_location='cpu')['model'], strict=False)删除['model']后成功运行

参考2021/5/24 voc-test数据集 yolov4测试 - 平台梦 (pingtaimeng.com)

你可能感兴趣的:(python,pycharm)