首页 Paddle框架 帖子详情
paddle优化器可以同时优化两个网络么? 已解决
收藏
快速回复
Paddle框架 问答模型训练 1298 4
paddle优化器可以同时优化两个网络么? 已解决
收藏
快速回复
Paddle框架 问答模型训练 1298 4

我一个模型里有两个网络使用同一个loss进行优化,这两个网络可以公用一个·优化器么?如果可以,是在优化器的parameters参数里指定么?

TC.Long
已解决
2# 回复于2021-06
可以的,可以将model1 和 model2的参数分别导出到 list1 与 list2 中,然后将 list1 与 list2 相加,传递给 parameters参数里就可以; 这部分可以参考:https://github.com/miraiwk/UGATIT-paddle/blob/master/UGATIT.py#L139-L148  
展开
0
收藏
回复
全部评论(4)
时间顺序
TC.Long
#2 回复于2021-06

可以的,可以将model1 和 model2的参数分别导出到 list1 与 list2 中,然后将 list1 与 list2 相加,传递给 parameters参数里就可以; 这部分可以参考:https://github.com/miraiwk/UGATIT-paddle/blob/master/UGATIT.py#L139-L148  

0
回复
FutureSI
#3 回复于2021-06
TC.Long #2
可以的,可以将model1 和 model2的参数分别导出到 list1 与 list2 中,然后将 list1 与 list2 相加,传递给 parameters参数里就可以; 这部分可以参考:https://github.com/miraiwk/UGATIT-paddle/blob/master/UGATIT.py#L139-L148  
展开

哦,原来这样就可以了。谢谢大佬!

0
回复
FutureSI
#4 回复于2021-06

少加一个优化器应该能省些内存吧

0
回复
FutureSI
#5 回复于2021-06

原来我是直接用一个list包了两个模型的parameters,试过了不行。原来把两个parameters的list直接接一起就ok了

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