问题描述:
在使用自己的数据集A训练完模型以后,在预测阶段使用的数据集B训练好的权重加载到本次模型当中,出现以下报错
原因分析:
通过err_log可知,在最后一层odm_conf层,数据维度不匹配,odm_conf层是模型最后一层的分类层。数据集A的类别数是4,而数据B训练类别数是21,这样就会导致训练出来的模型输出的维度不匹配。
RuntimeError: Error(s) in loading state_dict for
最新推荐文章于 2024-04-11 09:15:37 发布