torch.save()，torch.load()，state_dict()，load_state_dict()

这些函数是PyTorch中用于模型保存和加载的重要函数。下面是对它们的详细解析：

1. torch.save(obj, file):

   • 作用：将PyTorch模型保存到文件中。

   • 参数：

     • obj: 要保存的对象，可以是模型、张量或字典。
     • file: 要保存到的文件路径。

   • 示例：

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

2. torch.load(file):

   • 作用：从文件中加载保存的PyTorch模型。

   • 参数：

     • file: 要加载的文件路径。

   • 返回值：加载的对象。

   • 示例：

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

3. state_dict():

   • 作用：返回包含模型所有参数的字典对象。

   • 示例：

     model_state = model.state_dict()
     

4. load_state_dict(state_dict, strict=True):

   • 作用：加载预训练的参数字典到模型中。

   • 参数：

     • state_dict: 要加载的参数字典。
     • strict（可选）: 如果为True（默认值），则要求state_dict中的键与模型的参数名完全匹配。

   • 示例：

     model.load_state_dict(torch.load('pretrained.pth'))
     

这些函数在训练过程中非常有用，可以帮助保存模型的状态以及加载预训练的参数，使得模型的训练和部署更加方便。