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

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

# python # pycharm
Logo
智源数据社区

更多推荐

cover

解读自然语言处理：技术、应用与未来展望

avatar 智源数据社区

NLP-Beginner：自然语言处理入门练习----task 1基于机器学习的文本分类

NLP-Beginner：自然语言处理入门练习----task 1基于机器学习的文本分类

avatar 智源数据社区
cover

线下活动报名：智源研究院数据与行业应用 Workshop第三期

avatar 智源数据社区