如何忽略和初始化 state_dict 中的缺失键

Nea*_*bfi 3 python pytorch

我保存的 state_dict 不包含我模型中的所有层。如何忽略 state_dict 错误中的缺失键并初始化剩余的权重?

Nea*_*bfi 10

这可以通过传递strict=False到来实现load_state_dict

load_state_dict(state_dict, strict=False)
Run Code Online (Sandbox Code Playgroud)

文档