想要读取model的state_dict全部参数,并且修改state_dict,想将其中key等于某特定值的那一层参数进行替换操作。
state_dict = {k:v for k,v in model.state_dict().items()}
for k,v in state_dict:
if k == "target":
#执行操作
错误原因:字典只支持对k的遍历,所以还是需要加上.items()
修改后:
state_dict = {k:v for k,v in model.state_dict().items()}
for k,v in state_dict.items():
if k == "target":
#执行操作