Init0ne

摘要： 保存与加载整个模型 保存整个模型，包括网络结构和权重参数，保存后的文件用torch.load()加载后的类型是定义的网络结构类，如class CNN： torch.save(model, "model.pkl") 加载整个模型： model = torch.load("model.pkl") 只保存 阅读全文