【DL基础知识】比较两个模型参数值是否完全一样

DL基础知识
比较两个模型参数值是否完全一样

一、任务

• 有两个同样网络初始化得到的模型
  • model1
  • model2
• 逐层比较两个模型的参数是否相同

二、代码

model1_params = dict(model1.named_parameters())
model2_params = dict(model2.named_parameters())
for name, param1 in model1_params.items():
        if name not in model2_params:
            print(f"模型1存在参数{name},而模型2不存在")
        else:
            param2 = model2_params[name]
            if not torch.allclose(param1, param2):
                print(f"参数{name}不一致")

三、结果