预训练模型加载
预训练模型有两种保存方式
1.保存整个模型,即保存模型结构和参数
torch.save(net, PATH)
对应的加载方式:
torch.load(PATH)
2.只保存模型参数
torch.save(net.state_dict(), PATH)
对应的加载方式,读取时需要先搭建模型
model
预训练模型加载
预训练模型有两种保存方式
1.保存整个模型,即保存模型结构和参数
torch.save(net, PATH)
对应的加载方式:
torch.load(PATH)
2.只保存模型参数
torch.save(net.state_dict(), PATH)
对应的加载方式,读取时需要先搭建模型
model