这是权重文件中的某些层的数据类型(如浮点数精度)与模型中的对应层不一致 导致的
解决办法:在加载权重时使用 strict=False
参数来忽略不匹配的层
model.load_state_dict(weights)
改为:
model.load_state_dict(weights,strict=False)
这是权重文件中的某些层的数据类型(如浮点数精度)与模型中的对应层不一致 导致的
解决办法:在加载权重时使用 strict=False
参数来忽略不匹配的层
model.load_state_dict(weights)
改为:
model.load_state_dict(weights,strict=False)