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