torch模型的保存与加载

两种方式

'''
    第一种方式
    模型整体保存，占用空间会比较大
'''
torch.save(net, "../model/model.pkl")
torch.load("")

'''
    第二种方式
    保存模型参数，在加载模型参数之前，必须先建立模型
'''
torch.save(net.state_dict(), "params.pkl")
net.load_state_dict(torch.load("path_of_model_state_dict"))

下面是整个流程的代码

import torch

# data 数据加载
import numpy as np
import re  # regular expression

ff = open("../housing.data").readlines()
data = []
for item in ff:
    out = re.sub(r"\s{2,}", " ", item).strip()  # .strip()可去掉字符串前后的空格
    print