hello617的博客

原创 pytroch 如何保存与导入预训练模型

pytroch 如何保存与导入预训练模型1 如何保存预训练模型1.1 以字典形式 保存模型参数1.2 直接保存整个模型2 如何加载自己训练的预训练模型2.1 加载以字典形式保存的模型参数2.1.1 创建新的模型对象2.1.1 创建新的模型对象1 如何保存预训练模型1.1 以字典形式 保存模型参数torch.save可以保存我们的模型的部分参数 如下图。//n_hidden，n_layers...