torch.save怎么用

`torch.save` 是 PyTorch 中的一个函数，用于将一个或多个 PyTorch 对象保存到磁盘。这些对象可以是张量（tensor）、模型（model）的参数（state dict）、优化器（optimizer）的状态字典等。保存后的文件是二进制格式，通常具有 `.pt` 或 `.pth` 扩展名。

使用 `torch.save` 保存一个对象的基本语法如下：

import torch

# 假设你有一个模型的state_dict
model_state_dict = model.state_dict()

# 保存模型的state_dict到磁盘
torch.save(model_state_dict, 'model_state.pth')

如果你想要保存多个对象，比如模型的参数和优化器的状态，你可以将它们保存在一个字典中，然后保存这个字典：

# 假设你有模型的state_dict和优化器的状态字典
model_state_dict = model.state_dict()
optimizer_state_dict = optimizer.state_dict()

# 将它们保存在一个字典中
save_dict = {
    'model_state_dict': model_state_dict,
    'optimizer_state_dict': optimizer_state_dict
}

# 保存这个字典
torch.save(save_dict, 'model_and_optimizer.pth')