pytorch中net.eval() 和net.train()的使用

以前一直有個(gè)疑問(wèn)勃蜘,開始建立好模型,如果既沒(méi)有執(zhí)行net.train()也沒(méi)有執(zhí)行net.eval()時(shí)候模型處于哪種情況?

近天測(cè)試了一下,建立好的模型是處于 .train()模式下的:

建立好的模型

調(diào)試過(guò)程中查看網(wǎng)絡(luò)的 net.training 的值沐批。


調(diào)試查看

然后執(zhí)行下一步net.eval(),之后查看net.training 的屬性值:


執(zhí)行net.eval()

net.training的布爾值

然后執(zhí)行net.train()查看一下net.training的值:
執(zhí)行net.train()

執(zhí)行net.train()之后的net.training的布爾值

下面進(jìn)行一組對(duì)照試驗(yàn)蝎亚,驗(yàn)證網(wǎng)絡(luò)處于 train 或 eval 模式中的時(shí)候九孩,最后的出的結(jié)果是不一樣的 :

訓(xùn)練完train_datasets之后,model要來(lái)測(cè)試樣本了发框。在model(test_datasets)之前躺彬,需要加上model.eval(). 否則的話,有輸入數(shù)據(jù)梅惯,即使不訓(xùn)練宪拥,它也會(huì)改變權(quán)值。這是model中含有batch normalization層所帶來(lái)的的性質(zhì)铣减。
eval()時(shí)她君,pytorch會(huì)自動(dòng)把BN和DropOut固定住,不會(huì)取平均徙歼,而是用訓(xùn)練好的值犁河。
不然的話,一旦test的batch_size過(guò)小魄梯,很容易就會(huì)被BN層導(dǎo)致生成圖片顏色失真極大桨螺。eval()在非訓(xùn)練的時(shí)候是需要加的,沒(méi)有這句代碼酿秸,一些網(wǎng)絡(luò)層的值會(huì)發(fā)生變動(dòng)灭翔,不會(huì)固定,你神經(jīng)網(wǎng)絡(luò)每一次生成的結(jié)果也是不固定的,生成質(zhì)量可能好也可能不好肝箱。
這幾篇博客較好的解釋了droupout和BN層:
深度學(xué)習(xí)(二十九)Batch Normalization 學(xué)習(xí)筆記(贊贊)
Batch Normalization 簡(jiǎn)單理解

1.當(dāng)模型始終處于 train 模式中:

圖片.png

結(jié)果:


圖片.png

2.當(dāng)模型始終處于 eval 模式中:

圖片.png

結(jié)果:


圖片.png

3. 當(dāng)模型先處于eval 模式哄褒,后處于train 模式:

圖片.png

結(jié)果:


圖片.png

4. 當(dāng)模型先處于train 模式,后處于train 模式:

圖片.png

結(jié)果:


圖片.png
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末煌张,一起剝皮案震驚了整個(gè)濱河市呐赡,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌骏融,老刑警劉巖链嘀,帶你破解...
    沈念sama閱讀 211,290評(píng)論 6 491
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異档玻,居然都是意外死亡怀泊,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,107評(píng)論 2 385
  • 文/潘曉璐 我一進(jìn)店門误趴,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)霹琼,“玉大人,你說(shuō)我怎么就攤上這事凉当≡嫔辏” “怎么了?”我有些...
    開封第一講書人閱讀 156,872評(píng)論 0 347
  • 文/不壞的土叔 我叫張陵纤怒,是天一觀的道長(zhǎng)糯而。 經(jīng)常有香客問(wèn)我,道長(zhǎng)泊窘,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,415評(píng)論 1 283
  • 正文 為了忘掉前任像寒,我火速辦了婚禮烘豹,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘诺祸。我一直安慰自己携悯,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,453評(píng)論 6 385
  • 文/花漫 我一把揭開白布筷笨。 她就那樣靜靜地躺著憔鬼,像睡著了一般。 火紅的嫁衣襯著肌膚如雪胃夏。 梳的紋絲不亂的頭發(fā)上轴或,一...
    開封第一講書人閱讀 49,784評(píng)論 1 290
  • 那天,我揣著相機(jī)與錄音仰禀,去河邊找鬼照雁。 笑死,一個(gè)胖子當(dāng)著我的面吹牛答恶,可吹牛的內(nèi)容都是我干的饺蚊。 我是一名探鬼主播萍诱,決...
    沈念sama閱讀 38,927評(píng)論 3 406
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼污呼!你這毒婦竟也來(lái)了裕坊?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,691評(píng)論 0 266
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤燕酷,失蹤者是張志新(化名)和其女友劉穎碍庵,沒(méi)想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體悟狱,經(jīng)...
    沈念sama閱讀 44,137評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡静浴,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,472評(píng)論 2 326
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了挤渐。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片苹享。...
    茶點(diǎn)故事閱讀 38,622評(píng)論 1 340
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖浴麻,靈堂內(nèi)的尸體忽然破棺而出得问,到底是詐尸還是另有隱情,我是刑警寧澤软免,帶...
    沈念sama閱讀 34,289評(píng)論 4 329
  • 正文 年R本政府宣布宫纬,位于F島的核電站,受9級(jí)特大地震影響膏萧,放射性物質(zhì)發(fā)生泄漏漓骚。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,887評(píng)論 3 312
  • 文/蒙蒙 一榛泛、第九天 我趴在偏房一處隱蔽的房頂上張望蝌蹂。 院中可真熱鬧,春花似錦曹锨、人聲如沸孤个。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,741評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)齐鲤。三九已至,卻和暖如春椒楣,著一層夾襖步出監(jiān)牢的瞬間给郊,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,977評(píng)論 1 265
  • 我被黑心中介騙來(lái)泰國(guó)打工撒顿, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留丑罪,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 46,316評(píng)論 2 360
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像吩屹,于是被迫代替她去往敵國(guó)和親跪另。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,490評(píng)論 2 348

推薦閱讀更多精彩內(nèi)容