加载预训练的时候提示:
KeyError: "CascadeRCNN: 'backbone.layers.0.blocks.0.attn.relative_position_bias_table'"
这是因为预训练模型加载错误。。
应该加载ImageNet
上预训练的模型,而不是coco
数据集的预训练模型。
本文最后更新于2021年10月24日,已超过 1 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!
加载预训练的时候提示:
KeyError: "CascadeRCNN: 'backbone.layers.0.blocks.0.attn.relative_position_bias_table'"
这是因为预训练模型加载错误。。
应该加载ImageNet
上预训练的模型,而不是coco
数据集的预训练模型。
如果你对这篇文章有什么疑问或建议,欢迎下面留言提出,我看到会立刻回复!
打赏