[Python知识库]Python编程总结（二）

1. pytorch中保存好加载网络及参数的代码：

? ? a. 保存及加载整个网络的结构及权重信息：

? ? ? ? torch.save(Target_model, "model_save.pth");

? ? ? ? Target_model=torch.load("model_save.pth");

? ? b. 只保存网络的权重参数：

? ? ? ? torch.save(Target_model.state_dict, "model_save.pth");

? ? ? ? Target_model.load(torch.load("model_save.pth"));

2.?