预训练模型加载

预训练模型加载
预训练模型有两种保存方式
1.保存整个模型，即保存模型结构和参数

torch.save(net, PATH)

对应的加载方式：

torch.load(PATH)

2.只保存模型参数

torch.save(net.state_dict(), PATH)

对应的加载方式，读取时需要先搭建模型

model