加载预训练模型时KeyError: ‘model‘
加载预训练模型时KeyError: 'model'
·
self.load_state_dict(torch.load(pretrained, map_location='cpu')['model'], strict=False)删除['model']后成功运行
更多推荐
已为社区贡献1条内容
加载预训练模型时KeyError: 'model'
self.load_state_dict(torch.load(pretrained, map_location='cpu')['model'], strict=False)删除['model']后成功运行
更多推荐
所有评论(0)