Bolly_He的博客

目录 只保存模型训练参数 保存完整的模型 只保存模型训练参数 # 保存 torch.save(model.state_dict(), '\parameter.pkl') # 加载 model = ModelClass(...) model.load_state_dict(torch.load('\parameter.pkl')) 保存完整的模型 # 保存 torch.save(model, '\model.pkl') # 加载 model = torch.load('\model.pk.