model.save('EMD-H.h5') 改用torch.save

如果你使用 PyTorch 框架训练了模型，那么可以使用 PyTorch 提供的 `torch.save` 函数来保存模型。`torch.save` 函数可以将 PyTorch 模型保存为 `state_dict` 或者整个模型，具体可以根据需要选择。

以下是将整个模型保存为 `.pt` 文件的示例代码：

import torch

# 定义模型
model = ...

# 保存模型
torch.save(model, 'model.pt')

# 加载模型
loaded_model = torch.load('model.pt')

如果你只想保存模型的参数，而不保存整个模型，可以使用 `state_dict` 来保存模型。以下是将模型参数保存为 `.pt` 文件的示例代码：

import torch

# 定义模型
model = ...

# 保存模型参数
torch.save(model.state_dict(), 'model_params.pt')

# 加载模型参数
loaded_model = ...
loaded_model.load_state_dict(torch.load('model_params.pt'))

使用 `state_dict` 来保存模型参数可以使得模型文件更加轻量级，并且可以方便地进行模型的迁移和部署。