如何将pth文件中的参数导出

您可以使用 PyTorch 提供的 `state_dict()` 方法将模型参数导出到一个字典中，然后使用 Python 的 `pickle` 模块将字典保存为一个 pth 文件。

以下是一个示例代码：

import torch
import pickle

# 加载预训练模型
model = torch.load('model.pth')

# 导出参数到一个字典中
state_dict = model.state_dict()

# 保存字典为 pth 文件
with open('params.pth', 'wb') as f:
    pickle.dump(state_dict, f)

此时，您将得到一个名为 `params.pth` 的文件，其中包含了模型的所有参数。