部分网络关键字 ---- 完全匹配的情况load_state_dict 函数添加 参数 strict=True, 它直接忽略那些没有的dict,有相同的就复制,没有就直接放弃赋值!他要求预训练模型的关键字必须确切地严格地和 网络的 state_dict() 函数返回的关键字相匹配才能赋值。strict 也不是很智能,适用于那些 网络关键字 基本能够匹配的情况。否则即使加载成功,网络参数也是空的。

pytorch 实现冻结部分参数训练另一部分

pytorch 实现冻结部分参数训练另一部分

pytorch 实现冻结部分参数训练另一部分

pytorch 实现冻结部分参数训练另一部分