Paddle如何固定一个模型的参数?
收藏
我想训练一个LSTM+GAN的组合模型, 其中先训练GAN模型,再训练LSTM模型。
在Tensorflow中,把LSTM的输出直接传入训练好的GAN模型,指定opt.apply_gradients的作用在LSTM模型即可。如图:
但是在paddle中,好像并不支持这种操作,我使用类似的逻辑进行训练时:
报错行时时pre_loss.backward(),弹出了如下错误:
请问,paddle要实现先训练一个模型再训练另一个模型的逻辑,该怎么完成?
0
收藏
请登录后评论
paddle里通过优化器Optimizer的parameters=参数指定作用的模型参数。你说的那种情况或许可以用多个优化器解决,就像训练gan模型时一样
Optimizer的使用参考:
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/optimizer/Optimizer_cn.html#optimizer