pytorch模型加载与保存,并在保存的模型基础上进行训练
两种方法:1.只保存参数(官方推荐,消耗的存储空间较小),详细讲解此方法。2.保存整个模型结构一.只保存参数1.保存:方法一:torch.save(model.state_dict(), path)model:定义的模型的实例变量,如model = resnet(),path是模型保存的路径,如path = “./model.pth”,path = “./model.pkl”,path = “./model.tar”,一定要加上后缀。方法二:如果想保存某一次的训练参数和模型,那么可以使用