网络编程
位置:首页>> 网络编程>> Python编程>> pytorch 实现在一个优化器中设置多个网络参数的例子

pytorch 实现在一个优化器中设置多个网络参数的例子

作者:慢行厚积  发布时间:2021-11-30 17:12:05 

标签:pytorch,优化器,网络,参数

我就废话不多说了,直接上代码吧!

其实也不难,使用tertools.chain将参数链接起来即可


import itertools
...
self.optimizer = optim.Adam(itertools.chain(self.encoder.parameters(), self.decoder.parameters()), lr=self.opt.lr, betas=(self.opt.beta1, 0.999))
...

来源:https://www.cnblogs.com/wanghui-garcia/p/11447240.html

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com