亚洲乱码中文字幕综合,中国熟女仑乱hd,亚洲精品乱拍国产一区二区三区,一本大道卡一卡二卡三乱码全集资源,又粗又黄又硬又爽的免费视频

pytorch 實現(xiàn)在一個優(yōu)化器中設置多個網(wǎng)絡參數(shù)的例子

 更新時間:2020年02月20日 13:44:23   作者:慢行厚積  
今天小編就為大家分享一篇pytorch 實現(xiàn)在一個優(yōu)化器中設置多個網(wǎng)絡參數(shù)的例子,具有很好參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧

我就廢話不多說了,直接上代碼吧!

其實也不難,使用tertools.chain將參數(shù)鏈接起來即可

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 實現(xiàn)在一個優(yōu)化器中設置多個網(wǎng)絡參數(shù)的例子就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關文章

最新評論