淘先锋技术网

首页 1 2 3 4 5 6 7

保存格式?

pth在pytorch进行模型保存的时候,一般有两种保存格式,一种是保存整个模型,另一种是只保存模型的参数。torch.save(model.state_dict(), "my_model.pth") # 只保存模型的参数,torch.save(model, "my_model.pth") # 保存整个模型保存的模型参数实际上一个字典类型,通过key-value的形式来存储模型的所有参数。