1. 程式人生 > 程式設計 >pytorch 實現在一個優化器中設定多個網路引數的例子

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 實現在一個優化器中設定多個網路引數的例子就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。