欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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

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

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

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

相關(guān)文章

最新評論