优化器如何绑定多个模块的参数??
收藏
# pytorch中的写法:
torch.optim.SGD([module_1.parameters(), module_2.parameters()])
# 我的paddle代码,这样子不会报错,但不确定是否使用方法正确???
paddle.optimizer.SGD(module_1.parameters() + module_2.parameters())
我也知道可以把多个模组打包进一个类里面然后再绑定优化器,但我现在的业务不能将两个模组写进一个类里,这要如何将两个模组的参数绑定同个优化器??
0
收藏
请登录后评论
parameter参数的类型是列表,放进列表里试试
参考文档
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/optimizer/sgd/SGD_cn.html#sgd