Pytorch : too many values to unpack (expected 2)

想要读取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":
        #执行操作

猜你喜欢

转载自blog.csdn.net/macidoo/article/details/89279447