PyTorch 模型保存教程：使用 torch.save() 保存模型

在 PyTorch 中，保存模型可以使用 torch.save() 函数来实现。该函数的第一个参数是要保存的模型，第二个参数是保存模型的文件路径。

以下是保存模型的示例代码：

import torch

# 定义一个模型
class MyModel(torch.nn.Module):
    def __init__(self):
        super(MyModel, self).__init__()
        self.linear = torch.nn.Linear(1, 1)
    
    def forward(self, x):
        return self.linear(x)

model = MyModel()

# 保存模型
torch.save(model.state_dict(), 'my_model.pth')

在该示例代码中，我们定义了一个简单的模型 MyModel，并将其保存到文件 my_model.pth 中。要保存模型的状态字典，可以使用 model.state_dict() 函数。