pytorch保存模型与调用

在PyTorch中，保存和加载模型是一个常见的任务，这使得你可以将训练好的模型保存起来，以便在后续使用时无需从头开始训练。以下是基本的步骤：

1. **保存模型**:
- 使用`torch.save()`函数可以保存整个模型（包括结构和权重）到文件。例如：

     torch.save(model.state_dict(), 'model.pth')

如果你只需要保存模型的结构，可以使用`torch.jit.save()`来保存一个 traced 模型，它会包含模型的计算图：

     script_model = torch.jit.script(model)
     script_model.save('model.pt')

2. **加载模型**:
- 要加载已保存的模型，可以使用`torch.load()`函数读取文件，并结合模型实例创建完整的模型：

     model = YourModelClass()
     model.load_state_dict(torch.load('model.pth'))

对于JIT保存的模型，直接运行并加上`.eval()`模式即可：

     loaded_model = torch.jit.load('model.pt').eval()