pytorch:model.train和model.eval用法及區(qū)別詳解
使用PyTorch進(jìn)行訓(xùn)練和測(cè)試時(shí)一定注意要把實(shí)例化的model指定train/eval，eval（）時(shí)，框架會(huì)自動(dòng)把BN和DropOut固定住，不會(huì)取平均，而是用訓(xùn)練好的值，不然的話，一旦test的batch_size過小，很容易就會(huì)被BN層導(dǎo)致生成圖片顏色失真極大?。。。。?！
Class Inpaint_Network() ...... Model = Inpaint_Nerwoek() #train: Model.train(mode=True) ..... #test: Model.eval()
model.train()
啟用 BatchNormalization 和 Dropout
model.eval()
不啟用 BatchNormalization 和 Dropout
訓(xùn)練完train樣本后，生成的模型model要用來(lái)測(cè)試樣本。在model(test)之前，需要加上model.eval()，否則的話，有輸入數(shù)據(jù)，即使不訓(xùn)練，它也會(huì)改變權(quán)值。這是model中含有batch normalization層所帶來(lái)的的性質(zhì)。
在做one classification的時(shí)候，訓(xùn)練集和測(cè)試集的樣本分布是不一樣的，尤其需要注意這一點(diǎn)。
以上這篇pytorch:model.train和model.eval用法及區(qū)別詳解就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
下面小編就為大家?guī)?lái)一篇基于python爬蟲數(shù)據(jù)處理(詳解)。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來(lái)看看吧2017-06-06
Python中for循環(huán)語(yǔ)句實(shí)戰(zhàn)案例
這篇文章主要給大家介紹了關(guān)于Python中for循環(huán)語(yǔ)句的相關(guān)資料,python中for循環(huán)一般用來(lái)迭代字符串,列表,元組等,當(dāng)for循環(huán)用于迭代時(shí)不需要考慮循環(huán)次數(shù),循環(huán)次數(shù)由后面的對(duì)象長(zhǎng)度來(lái)決定,需要的朋友可以參考下2023-09-09
這篇文章主要介紹了Python模塊zipfile原理及使用方法詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-08-08
這篇文章主要介紹了python中的yield使用方法,需要的朋友可以參考下2014-02-02
這篇文章主要介紹了python裝飾器的三種裝飾模式，幫助大家更好的理解和學(xué)習(xí)python 裝飾器，感興趣的朋友可以了解下2020-09-09
Django imgareaselect手動(dòng)剪切頭像實(shí)現(xiàn)方法
這篇文章主要介紹了Django imgareaselect手動(dòng)剪切頭像實(shí)現(xiàn)方法,實(shí)例分析了Django框架操作圖片的相關(guān)技巧,需要的朋友可以參考下2015-05-05
教你使用Pandas直接核算Excel中的快遞費(fèi)用
文中仔細(xì)說(shuō)明了怎么根據(jù)賬單核算運(yùn)費(fèi).首先要確定運(yùn)費(fèi)規(guī)則,然后根據(jù)運(yùn)費(fèi)規(guī)則編寫代碼,生成核算列（快遞費(fèi) = 省份*重量）,最后輸入賬單,進(jìn)行核算.將腳本件生成EXE文件,就可以使用啦,需要的朋友可以參考下2021-05-05
Python如何獲取當(dāng)前路徑并列出當(dāng)前路徑下的所有文件
這篇文章主要介紹了Python如何獲取當(dāng)前路徑并列出當(dāng)前路徑下的所有文件問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-06-06
這篇文章主要介紹了Python面向?qū)ο笾惖亩x與繼承用法,結(jié)合實(shí)例形式分析了Python類的定義、實(shí)例化、繼承等基本操作技巧,需要的朋友可以參考下2019-01-01
本文主要介紹了Django中外鍵使用總結(jié)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-07-07