PaddlePaddle中怎么保存和加載訓(xùn)練好的模型

小億
206
2024-03-19 15:36:38

在PaddlePaddle中，可以使用paddle.save和paddle.load來(lái)保存和加載訓(xùn)練好的模型。

  1. 保存模型：
# 保存模型
paddle.save(model.state_dict(), 'model.pdparams')
  1. 加載模型：
# 加載模型
model_dict = paddle.load('model.pdparams')
model.set_dict(model_dict)

其中，model.state_dict()用于獲取模型的參數(shù)字典，paddle.save用于保存參數(shù)字典到文件中，paddle.load用于加載參數(shù)字典。需要注意的是，在加載模型之后，還需要將參數(shù)字典設(shè)置給模型，即model.set_dict(model_dict)。

0 踩