首页 Paddle框架 帖子详情
优化器如何绑定多个模块的参数??
收藏
快速回复
Paddle框架 问答模型训练炼丹技巧 973 2
优化器如何绑定多个模块的参数??
收藏
快速回复
Paddle框架 问答模型训练炼丹技巧 973 2
# pytorch中的写法:
torch.optim.SGD([module_1.parameters(), module_2.parameters()])
# 我的paddle代码,这样子不会报错,但不确定是否使用方法正确???
paddle.optimizer.SGD(module_1.parameters() + module_2.parameters())

我也知道可以把多个模组打包进一个类里面然后再绑定优化器,但我现在的业务不能将两个模组写进一个类里,这要如何将两个模组的参数绑定同个优化器??

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

parameter参数的类型是列表,放进列表里试试

0
回复
AIStudio810258
#3 回复于2021-05

参考文档

https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/optimizer/sgd/SGD_cn.html#sgd

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