Nea*_*bfi 3 python pytorch
我保存的 state_dict 不包含我模型中的所有层。如何忽略 state_dict 错误中的缺失键并初始化剩余的权重?
Nea*_*bfi 10
这可以通过传递strict=False到来实现load_state_dict。
strict=False
load_state_dict
load_state_dict(state_dict, strict=False)
文档
归档时间:
5 年,5 月 前
查看次数:
3140 次
最近记录:
4 年,7 月 前