Pytorch如何保存训练好的模型

1、（1）只保存模型参数字典（推荐）代码：#保存torch.save(the_model.state_dict(), PATH)#读取the_model = TheModelClass(*args, **kwargs)the_model.load_state_dict(torch.load(PATH))

2、（2）保存整个模型代码：#保存torch.save(the_model, PATH)#读取the_model = torch.load(PATH)