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ù)的例子就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
在Windows系統(tǒng)上搭建Nginx+Python+MySQL環(huán)境的教程
這篇文章主要介紹了在Windows系統(tǒng)上搭建Nginx+Python+MySQL環(huán)境的教程,文中使用flup中間件及FastCGI方式連接,需要的朋友可以參考下2015-12-12對python插入數(shù)據(jù)庫和生成插入sql的示例講解
今天小編就為大家分享一篇對python插入數(shù)據(jù)庫和生成插入sql的示例講解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11Win10系統(tǒng)下安裝labelme及json文件批量轉化方法
這篇文章主要介紹了Win10系統(tǒng)下安裝labelme及json文件批量轉化的方法,文中較詳細的給大家介紹了安裝過程 ,需要的朋友可以參考下2019-07-07解決Alexnet訓練模型在每個epoch中準確率和loss都會一升一降問題
這篇文章主要介紹了解決Alexnet訓練模型在每個epoch中準確率和loss都會一升一降問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06Python PyQt5干貨滿滿小項目輕松實現(xiàn)高效摳圖去背景
PyQt5以一套Python模塊的形式來實現(xiàn)功能。它包含了超過620個類,600個方法和函數(shù)。本篇文章手把手帶你用PyQt5輕松實現(xiàn)圖片扣除背景,大家可以在過程中查缺補漏,提升水平2021-11-11Python判斷一個三位數(shù)是否為水仙花數(shù)的示例
今天小編就為大家分享一篇Python判斷一個三位數(shù)是否為水仙花數(shù)的示例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11