在PaddlePaddle中,可以使用paddle.save
和paddle.load
來保存和加載訓練好的模型。
# 保存模型
paddle.save(model.state_dict(), 'model.pdparams')
# 加載模型
model_dict = paddle.load('model.pdparams')
model.set_dict(model_dict)
其中,model.state_dict()
用于獲取模型的參數字典,paddle.save
用于保存參數字典到文件中,paddle.load
用于加載參數字典。需要注意的是,在加載模型之后,還需要將參數字典設置給模型,即model.set_dict(model_dict)
。