如何调用保存好的pytorch模型

您可以使用PyTorch中的torch.load()函数来加载已保存的模型。您需要将模型的路径和名称传递给该函数。示例代码如下：

import torch

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

# 实例化模型
model = MyModel()

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

# 加载模型
loaded_model = MyModel()
loaded_model.load_state_dict(torch.load('model.pth'))

请注意，加载模型的代码必须与保存模型的代码位于同一程序中。如果您需要在不同的程序中加载模型，则需要将模型状态字典保存为.pth或.pt文件，并将该文件复制到要加载模型的程序中。