Pytorch evaluation每次運(yùn)行結(jié)果不同的解決
這兩天跑測(cè)試圖時(shí),發(fā)現(xiàn)用同樣的model,同樣的測(cè)試圖,每次運(yùn)行結(jié)果不同;
經(jīng)過漫長的debug發(fā)現(xiàn),在net architure中有dropout,如下(4):
(conv_block): Sequential( (0): ReflectionPad2d((1, 1, 1, 1)) (1): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1)) (2): InstanceNorm2d(256, eps=1e-05, momentum=0.1, affine=False) (3): ReLU(inplace) (4): Dropout(p=0.5) (5): ReflectionPad2d((1, 1, 1, 1)) (6): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1)) (7): InstanceNorm2d(256, eps=1e-05, momentum=0.1, affine=False) )
在跑evaluation的時(shí)候,因?yàn)閐ropout的存在,每次運(yùn)行會(huì)隨機(jī)丟一些中間結(jié)果,從而導(dǎo)致最終結(jié)果有差異;
可以在evaluation過程中,使用eval() class強(qiáng)制丟掉random的內(nèi)容,code如下:
self.fake_B = self.netG.eval().forward(self.real_A)
以上這篇Pytorch evaluation每次運(yùn)行結(jié)果不同的解決就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
用Python編寫一個(gè)簡單的FUSE文件系統(tǒng)的教程
這篇文章主要介紹了用Python編寫一個(gè)簡單的FUSE文件系統(tǒng)的教程,對(duì)于數(shù)據(jù)的備份很有幫助,需要的朋友可以參考下2015-04-04Python+wxPython構(gòu)建中文OCR截圖工具
隨著數(shù)字化時(shí)代的到來,我們經(jīng)常需要從圖像中提取文本信息,本文將介紹如何使用Python編程語言和wxPython圖形用戶界面庫構(gòu)建一個(gè)簡單的中文OCR截圖工具,需要的可以參考下2023-08-08python 識(shí)別登錄驗(yàn)證碼圖片功能的實(shí)現(xiàn)代碼(完整代碼)
這篇文章主要介紹了python 識(shí)別登錄驗(yàn)證碼圖片功能,本文通過實(shí)例代碼給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-07-07