分类：pytorch模型保存

2021
08-08

pytorch模型的保存和加载、checkpoint操作

其实之前笔者写代码的时候用到模型的保存和加载，需要用的时候就去度娘搜一下大致代码，现在有时间就来整理下整个pytorch模型的保存和加载，开始学习把~pytorch的模型和参数是分开的，可以分别保存或加载模型和参数。所以pytorch的保存和加载对应存在两种方式：1.直接保存加载模型（1）保存和加载整个模型#保存模型torch.save(model,'model.pth\pkl\pt')#一般形式torch.save(net,PATH)#加载模型model=torch.load('mode... 继续阅读 >