一、模型的存储与读取
主要涉及到torch.save和torch.load函数
新建两个python文件:
1.在model_save文件中保存模型(方式一)和模型参数(方式二)
2.在model_load文件中读取模型(方式一)和模型参数并装载模型(方式二)
一、模型的存储与读取
主要涉及到torch.save和torch.load函数
新建两个python文件:
1.在model_save文件中保存模型(方式一)和模型参数(方式二)
2.在model_load文件中读取模型(方式一)和模型参数并装载模型(方式二)