pytorch训练后怎么保存

在 PyTorch 中，我们可以使用 `torch.save()` 函数将训练好的模型保存到磁盘上。具体的使用方法如下：

import torch

# 定义模型
model = YourModel()

# 训练模型

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

在上面的代码中，我们首先定义了一个模型 `model`，然后进行训练，最后通过 `torch.save()` 函数将模型的参数保存到 `model.pt` 文件中。

如果您想要将整个模型（包括模型结构）保存到文件中，可以使用 `torch.save(model, 'model.pt')`。这将会保存整个模型，并且可以通过 `torch.load('model.pt')` 加载模型。