保存格式？

pth在pytorch進行模型保存的時候，一般有兩種保存格式，一種是保存整個模型，另一種是只保存模型的參數(shù)。torch.save(model.state_dict(), "my_model.pth") # 只保存模型的參數(shù)，torch.save(model, "my_model.pth") # 保存整個模型保存的模型參數(shù)實際上一個字典類型，通過key-value的形式來存儲模型的所有參數(shù)。