pytorch只想对模型加载部分参数

前言

对一个torch模型保存可以使用save dict操作，其本质是一个字典，字典的k是string，表示参数所在的层，v是此层的参数。可以自己debug看看
对于模型来说，load_state_dict可以加载之前保存的参数字典，此函数有一个参数，是是否严格加载参数。

model.load_state_dict(related_params, strict=True)

默认strict是False，当权重中的key和网络中匹配就加载，不匹配就跳过。
如果strict是True，那必须完全匹配，不然就报错。
默认strict是False模式的。

方法

那么，可以直接用false模式的load state dict，只保存想要的参数，然后用整个模型去load state dict，这样就可以只加载部分层的参数了。
另外，保存过的state dict也是可以自己改动的，

related_params={k:v for k,v in params.items() if 'mix' in k}

像这样子只把想保存的参数截出来，在用整个模型load state dict也可以。