pytorch掉坑記錄:model.eval的作用說明
訓(xùn)練完train_datasets之后,model要來測試樣本了。在model(test_datasets)之前,需要加上model.eval(). 否則的話,有輸入數(shù)據(jù),即使不訓(xùn)練,它也會改變權(quán)值。
這是model中含有batch normalization層所帶來的的性質(zhì)。
在做one classification的時候,訓(xùn)練集和測試集的樣本分布是不一樣的,尤其需要注意這一點。
補充知識:pytorch測試的時候為何要加上model.eval()
Do need to use model.eval() when I test?
Sure, Dropout works as a regularization for preventing overfitting during training.
It randomly zeros the elements of inputs in Dropout layer on forward call.
It should be disabled during testing since you may want to use full model (no element is masked)
使用PyTorch進行訓(xùn)練和測試時一定注意要把實例化的model指定train/eval,eval()時,框架會自動把BN和DropOut固定住,不會取平均,而是用訓(xùn)練好的值,不然的話,一旦test的batch_size過小,很容易就會被BN層導(dǎo)致生成圖片顏色失真極大?。。。。?!
以上這篇pytorch掉坑記錄:model.eval的作用說明就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
使用C語言來擴展Python程序和Zope服務(wù)器的教程
這篇文章主要介紹了使用C語言來擴展Python程序和Zope服務(wù)器的教程,本文來自于IBM官方網(wǎng)站技術(shù)文檔,需要的朋友可以參考下2015-04-04
利用python3篩選excel中特定的行(行值滿足某個條件/行值屬于某個集合)
這篇文章主要給大家介紹了關(guān)于利用python3篩選excel中特定的行(行值滿足某個條件/行值屬于某個集合)的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
Python?pyecharts數(shù)據(jù)可視化實例詳解
PyEcharts是一個用于生成?Echarts圖表的類庫,?Python是一門富有表達力的語言,很適合用于數(shù)據(jù)處理,下面這篇文章主要給大家介紹了關(guān)于Python?pyecharts數(shù)據(jù)可視化的相關(guān)資料,需要的朋友可以參考下2022-05-05
Python scrapy爬取起點中文網(wǎng)小說榜單
爬蟲的基礎(chǔ)內(nèi)容已經(jīng)全部學(xué)玩,博主決定想著更加標(biāo)準(zhǔn)化以及實用能力更強的scrapy進發(fā),今天記錄自己第一個scrapy爬蟲項目. scrapy爬取起點中文網(wǎng)24小時熱銷榜單,需要的朋友可以參考下2021-06-06
數(shù)據(jù)清洗--DataFrame中的空值處理方法
今天小編就為大家分享一篇數(shù)據(jù)清洗--DataFrame中的空值處理方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-07-07

