正文
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))
...
