pytorch查看模型参数两

在PyTorch中，查看模型参数通常涉及两个主要步骤：保存模型参数和加载模型以访问这些参数。

1. **保存模型参数**[^2]:

   torch.save(model.state_dict(), "my_model.pth")  # 这里"model"是你想要保存参数的模型实例

`state_dict()`函数返回模型的参数字典，`save()`方法用于持久化存储这些参数。

2. **查看/加载模型参数**[^1]:
要查看保存的参数，你可以直接加载模型并打印其状态字典：

   loaded_state_dict = torch.load("my_model.pth")
   print(loaded_state_dict)  # 打印模型参数的详细结构

如果你想只加载模型而不重新构建整个模型，可以这样做：

   new_model = YourModelClass()  # 假设YourModelClass是你的模型类
   new_model.load_state_dict(loaded_state_dict)