一、模型的保存與加載
實現訓練過程中模型的保存,以及在預訓練的基礎上繼續訓練模型
①保存和加載整個模型
# 保存和加載整個模型
torch.save(model_object, 'model.pkl')
model = torch.load('model.pkl')
②只保存模型中的參數
# 僅保存和加載模型參數(推薦使用)
torch.save(model_object.state_dict(), 'params.pkl')
model_object.load_state_dict(torch.load('params.pkl'))