PyTorch深度学习模型的保存和加载流程详解

一、模型参数的保存和加载  torch.save(module.state_dict(), path)：使用module.state_dict()函数获取各层已经训练好的参数和缓冲区，然后将参数和缓冲区保存到path所指定的文件存放路径（常用文件格式为.pt、