pytorch 把训练好的pth文件导出到mat文件里

可以使用Python中的SciPy库将Pytorch模型的权重保存为MATLAB的.mat文件。下面是一个简单的示例代码：

import scipy.io as sio  
import torch 

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

# 将模型的权重转换为numpy数组
weights = {}
for name, param in model.named_parameters():
    weights[name] = param.cpu().detach().numpy()

# 保存权重为MATLAB格式
sio.savemat('trained_model.mat', weights)

这将生成一个名为“trained_model.mat”的MATLAB格式文件，其中包含模型的权重。请确保在运行代码之前安装SciPy库。