今天小编就为大家分享一篇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)) ...
以上就是pytorch 实现在一个优化器中设置多个网络参数的例子的详细内容，更多请关注0133技术站其它相关文章！
