pytorch 保存模型信息

在 PyTorch 中，可以使用 `torch.save()` 函数将模型的参数保存到文件中。具体步骤如下：

1. 定义模型并训练模型；
2. 将模型的参数保存到文件中，例如：

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

其中，`model.state_dict()` 返回模型的参数字典，`'model.pth'` 是保存的文件名。