首页 Paddle框架 帖子详情
高层API恢复训练问题
收藏
快速回复
Paddle框架 问答模型训练 375 3
高层API恢复训练问题
收藏
快速回复
Paddle框架 问答模型训练 375 3

为什么不能使用model.load( )恢复训练,依然是从头开始训练,到底是哪出现问题了呢,求教!!

0
收藏
回复
全部评论(3)
时间顺序
三岁
#2 回复于2021-08

能够把报错贴出来看一下嘛?

这方面再训练可以参考一下paddle高阶AIP课程,里面有详尽描述

0
回复
三先生01
#3 回复于2021-08
三岁 #2
能够把报错贴出来看一下嘛? 这方面再训练可以参考一下paddle高阶AIP课程,里面有详尽描述

谢谢回复,不是报错,是我想接着上次的第15轮继续训练,然后使用mode.load( )加载断点模型,但仍然是从头开始训练。

0
回复
d
docoter_c
#4 回复于2021-08

# example 1: dynamic graph
import paddle
emb = paddle.nn.Embedding(10, 10)
layer_state_dict = emb.state_dict()

# save state_dict of emb
paddle.save(layer_state_dict, "emb.pdparams")

scheduler = paddle.optimizer.lr.NoamDecay(
d_model=0.01, warmup_steps=100, verbose=True)
adam = paddle.optimizer.Adam(
learning_rate=scheduler,
parameters=emb.parameters())
opt_state_dict = adam.state_dict()

# save state_dict of optimizer
paddle.save(opt_state_dict, "adam.pdopt")
# save weight of emb
paddle.save(emb.weight, "emb.weight.pdtensor")

# load state_dict of emb
load_layer_state_dict = paddle.load("emb.pdparams")
# load state_dict of optimizer
load_opt_state_dict = paddle.load("adam.pdopt")
# load weight of emb
load_weight = paddle.load("emb.weight.pdtensor")
emb.weight.set_value(load_weight)

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户