pytorch如何保存和加载模型
两种方法：保存和加载参数 和 保存加载整个模型
保存和加载参数
#保存
torch.save(model.state_dict,PATH) #PATH推荐格式为.pt
#加载
model=TheModelClass(*args, **kwargs )
model.load_state_dict(torch.load(PATH) )保存加载整个模型
#保存
torch.save(model,PATH)
#加载
model = torch.load(PATH)