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