首页 Paddle框架 帖子详情
Paddle如何固定一个模型的参数?
收藏
快速回复
Paddle框架 问答深度学习模型训练 943 2
Paddle如何固定一个模型的参数?
收藏
快速回复
Paddle框架 问答深度学习模型训练 943 2

我想训练一个LSTM+GAN的组合模型, 其中先训练GAN模型,再训练LSTM模型。

在Tensorflow中,把LSTM的输出直接传入训练好的GAN模型,指定opt.apply_gradients的作用在LSTM模型即可。如图:

但是在paddle中,好像并不支持这种操作,我使用类似的逻辑进行训练时:

报错行时时pre_loss.backward(),弹出了如下错误:

请问,paddle要实现先训练一个模型再训练另一个模型的逻辑,该怎么完成?

 

0
收藏
回复
全部评论(2)
时间顺序
FutureSI
#2 回复于2021-10

paddle里通过优化器Optimizer的parameters=参数指定作用的模型参数。你说的那种情况或许可以用多个优化器解决,就像训练gan模型时一样

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