pytorch保存和加载训练模型torch.save()model.state_dict()torch.load()model.load_state_dictdel checkpoint
pytorch保存和加载训练模型torch.save()torch.load()del checkpointtorch.save()torch.save()可以保存一个字典,这里除了保存预训练模型的权重之外,还可以保存一些训练信息:torch.load()torch.load()可以加载一个字典:del checkpoint加载到模型之后,删掉原有的checkpoint以节省内存。del checkpoint # dereference seems crucial...