.h5文件转.pt文件

将.h5文件转换为.pt文件的步骤如下：

1. 安装PyTorch和h5py库：

pip install torch
pip install h5py

2. 加载.h5文件并转换为PyTorch模型：

import torch
import h5py

# 加载.h5文件
h5_model = h5py.File('model.h5', 'r')

# 创建PyTorch模型
pt_model = {}

for k, v in h5_model.items():
    pt_model[k] = torch.tensor(v)

# 保存为.pt文件
torch.save(pt_model, 'model.pt')

3. 验证.pt文件是否正确转换：

# 加载.pt文件
pt_model = torch.load('model.pt')

# 验证模型参数是否一致
for k, v in pt_model.items():
    print(k, v.size())

如果输出的大小与原始模型中的大小相同，则表示成功转换为PyTorch模型。