亚洲综合原千岁中文字幕_国产精品99久久久久久久vr_无码人妻aⅴ一区二区三区浪潮_成人h动漫精品一区二区三

主頁 > 知識庫 > 聊聊PyTorch中eval和no_grad的關系

聊聊PyTorch中eval和no_grad的關系

熱門標簽:陜西金融外呼系統 唐山智能外呼系統一般多少錢 白銀外呼系統 海南400電話如何申請 激戰2地圖標注 公司電話機器人 哈爾濱ai外呼系統定制 廣告地圖標注app 騰訊外呼線路

首先這兩者有著本質上區別

model.eval()是用來告知model內的各個layer采取eval模式工作。這個操作主要是應對諸如dropout和batchnorm這些在訓練模式下需要采取不同操作的特殊layer。訓練和測試的時候都可以開啟。

torch.no_grad()則是告知自動求導引擎不要進行求導操作。這個操作的意義在于加速計算、節約內存。但是由于沒有gradient,也就沒有辦法進行backward。所以只能在測試的時候開啟。

所以在evaluate的時候,需要同時使用兩者。

model = ...
dataset = ...
loss_fun = ...

# training
lr=0.001
model.train()
for x,y in dataset:
 model.zero_grad()
 p = model(x)
 l = loss_fun(p, y)
 l.backward()
 for p in model.parameters():
  p.data -= lr*p.grad
 
# evaluating
sum_loss = 0.0
model.eval()
with torch.no_grad():
 for x,y in dataset:
  p = model(x)
  l = loss_fun(p, y)
  sum_loss += l
print('total loss:', sum_loss)

另外no_grad還可以作為函數是修飾符來用,從而簡化代碼。

def train(model, dataset, loss_fun, lr=0.001):
 model.train()
 for x,y in dataset:
  model.zero_grad()
  p = model(x)
  l = loss_fun(p, y)
  l.backward()
  for p in model.parameters():
   p.data -= lr*p.grad
 
@torch.no_grad()
def test(model, dataset, loss_fun):
 sum_loss = 0.0
 model.eval()
 for x,y in dataset:
  p = model(x)
  l = loss_fun(p, y)
  sum_loss += l
 return sum_loss

# main block:
model = ...
dataset = ...
loss_fun = ...

# training
train()
# test
sum_loss = test()
print('total loss:', sum_loss)

補充:pytorch中model.train、model.eval以及torch.no_grad的用法

1、model.train()

啟用 BatchNormalization 和 Dropout

model.train() 讓model變成訓練模式,此時 dropout和batch normalization的操作在訓練起到防止網絡過擬合的問題

2、model.eval()

不啟用 BatchNormalization 和 Dropout

model.eval(),pytorch會自動把BN和DropOut固定住,而用訓練好的值。不然的話,一旦test的batch_size過小,很容易就會被BN層導致所生成圖片顏色失真極大

訓練完train樣本后,生成的模型model要用來測試樣本。在model(test)之前,需要加上model.eval(),否則的話,有輸入數據,即使不訓練,它也會改變權值。這是model中含有batch normalization層所帶來的的性質。

對于在訓練和測試時為什么要這樣做,可以從下面兩段話理解:

在訓練的時候, 會計算一個batch內的mean 和var, 但是因為是小batch小batch的訓練的,所以會采用加權或者動量的形式來將每個batch的 mean和var來累加起來,也就是說再算當前的batch的時候,其實當前的權重只是占了0.1, 之前所有訓練過的占了0.9的權重,這樣做的好處是不至于因為某一個batch太過奇葩而導致的訓練不穩定。

好,現在假設訓練完成了, 那么在整個訓練集上面也得到了一個最終的”mean 和var”, BN層里面的參數也學習完了(如果指定學習的話),而現在需要測試了,測試的時候往往會一張圖一張圖的去測,這時候沒有batch而言了,對單獨一個數據做 mean和var是沒有意義的, 那么怎么辦,實際上在測試的時候BN里面用的mean和var就是訓練結束后的mean_final 和 val_final. 也可說是在測試的時候BN就是一個變換。所以在用pytorch的時候要注意這一點,在訓練之前要有model.train() 來告訴網絡現在開啟了訓練模式,在eval的時候要用”model.eval()”, 用來告訴網絡現在要進入測試模式了.因為這兩種模式下BN的作用是不同的。

3、torch.no_grad()

這條語句的作用是:在測試時不進行梯度的計算,這樣可以在測試時有效減小顯存的占用,以免發生顯存溢出(OOM)。

這條語句通常加在網絡預測的那條代碼上。

4、pytorch中model.eval()和“with torch.no_grad()區別

兩者區別

在PyTorch中進行validation時,會使用model.eval()切換到測試模式,在該模式下,

主要用于通知dropout層和batchnorm層在train和val模式間切換

在train模式下,dropout網絡層會按照設定的參數p設置保留激活單元的概率(保留概率=p); batchnorm層會繼續計算數據的mean和var等參數并更新。

在val模式下,dropout層會讓所有的激活單元都通過,而batchnorm層會停止計算和更新mean和var,直接使用在訓練階段已經學出的mean和var值。

該模式不會影響各層的gradient計算行為,即gradient計算和存儲與training模式一樣,只是不進行反傳(backprobagation)

而with torch.zero_grad()則主要是用于停止autograd模塊的工作,以起到加速和節省顯存的作用,具體行為就是停止gradient計算,從而節省了GPU算力和顯存,但是并不會影響dropout和batchnorm層的行為。

使用場景

如果不在意顯存大小和計算時間的話,僅僅使用model.eval()已足夠得到正確的validation的結果;而with torch.zero_grad()則是更進一步加速和節省gpu空間(因為不用計算和存儲gradient),從而可以更快計算,也可以跑更大的batch來測試。

以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方,望不吝賜教。

您可能感興趣的文章:
  • pytorch掉坑記錄:model.eval的作用說明
  • pytorch:model.train和model.eval用法及區別詳解
  • 解決Pytorch中的神坑:關于model.eval的問題

標簽:黑龍江 常德 上海 鷹潭 黔西 惠州 四川 益陽

巨人網絡通訊聲明:本文標題《聊聊PyTorch中eval和no_grad的關系》,本文關鍵詞  聊聊,PyTorch,中,eval,和,grad,;如發現本文內容存在版權問題,煩請提供相關信息告之我們,我們將及時溝通與處理。本站內容系統采集于網絡,涉及言論、版權與本站無關。
  • 相關文章
  • 下面列出與本文章《聊聊PyTorch中eval和no_grad的關系》相關的同類信息!
  • 本頁收集關于聊聊PyTorch中eval和no_grad的關系的相關信息資訊供網民參考!
  • 推薦文章
    日韩一级黄色| 久草免费在线视频| 中文字幕一区二区三区精彩视频 | 欧美夜夜骑 青草视频在线观看完整版 久久精品99无色码中文字幕 欧美日韩一区二区在线观看视频 欧美中文字幕在线视频 www.99精品 香蕉视频久久 | 国产一区二区精品| 免费的黄视频| 成人免费观看的视频黄页| 欧美大片一区| 精品美女| 成人a级高清视频在线观看| 成人a级高清视频在线观看| 成人免费观看的视频黄页| 日日日夜夜操| 九九精品久久| 日本特黄特色aa大片免费| 精品国产亚一区二区三区| 免费的黄视频| 国产美女在线一区二区三区| 九九精品在线| 国产91精品一区| 国产伦理精品| 韩国三级视频网站| 91麻豆精品国产高清在线| 999精品视频在线| 久草免费在线色站| 毛片成人永久免费视频| 91麻豆爱豆果冻天美星空| 国产91丝袜在线播放0| 美女被草网站| 久久国产一区二区| 日韩av东京社区男人的天堂| 国产福利免费视频| 精品在线观看国产| 欧美夜夜骑 青草视频在线观看完整版 久久精品99无色码中文字幕 欧美日韩一区二区在线观看视频 欧美中文字幕在线视频 www.99精品 香蕉视频久久 | 香蕉视频一级| 亚欧成人毛片一区二区三区四区| 成人av在线播放| 欧美爱色| 国产综合91天堂亚洲国产| 精品国产亚洲一区二区三区| 一级女人毛片人一女人| 久久精品店| 国产成人女人在线视频观看 | 精品久久久久久综合网| 二级片在线观看| 美国一区二区三区| 国产伦精品一区二区三区无广告| 韩国三级香港三级日本三级la| 精品国产一区二区三区久久久狼| 日韩av东京社区男人的天堂| 日本久久久久久久 97久久精品一区二区三区 狠狠色噜噜狠狠狠狠97 日日干综合 五月天婷婷在线观看高清 九色福利视频 | 国产激情一区二区三区| 免费的黄色小视频| 国产视频网站在线观看| 99久久精品国产麻豆| 毛片电影网| 国产视频一区二区三区四区| 成人在免费观看视频国产| 久久99中文字幕| 欧美一区二区三区性| 国产网站免费| a级黄色毛片免费播放视频| 国产亚洲男人的天堂在线观看| 日韩一级黄色片| 免费国产一级特黄aa大片在线| 欧美日本免费| 天天做人人爱夜夜爽2020毛片| 国产a视频| 精品视频一区二区三区免费| 亚洲 男人 天堂| 日本特黄特色aaa大片免费| 国产伦理精品| 国产成+人+综合+亚洲不卡| 午夜在线影院| 欧美激情一区二区三区在线| 四虎影视久久久| 国产网站免费在线观看| 欧美另类videosbestsex久久| 色综合久久天天综合绕观看| 91麻豆精品国产自产在线| 欧美一级视频免费| 黄色福利片| 日韩字幕在线| 久久国产影院| 国产综合成人观看在线| 免费国产在线观看| 你懂的国产精品| 日韩专区在线播放| 国产视频久久久| 四虎影视库国产精品一区| 香蕉视频一级| 国产一区二区精品久久91| 亚洲 激情| 黄视频网站在线看| 久久精品免视看国产明星| 亚久久伊人精品青青草原2020| 成人高清视频在线观看| 国产成人精品影视| 91麻豆国产福利精品| 日日日夜夜操| 国产91精品一区| 欧美夜夜骑 青草视频在线观看完整版 久久精品99无色码中文字幕 欧美日韩一区二区在线观看视频 欧美中文字幕在线视频 www.99精品 香蕉视频久久 | 国产国产人免费视频成69堂| 九九精品久久| 91麻豆tv| 免费国产在线观看| 四虎论坛| 黄色免费网站在线| 四虎影视库国产精品一区| 精品国产亚洲一区二区三区| 国产精品自拍亚洲| 国产a毛片| 欧美一级视| 国产一区精品| 国产一区免费观看| 国产网站免费| 91麻豆精品国产片在线观看| 日韩在线观看视频网站| 黄视频网站免费看| 久久99中文字幕| 日本久久久久久久 97久久精品一区二区三区 狠狠色噜噜狠狠狠狠97 日日干综合 五月天婷婷在线观看高清 九色福利视频 | 美女免费黄网站| 国产不卡在线观看视频| 天天色色色| 欧美日本二区| 午夜欧美福利| 亚洲 国产精品 日韩| 日本伦理片网站| 欧美爱色| 91麻豆爱豆果冻天美星空| 国产视频一区二区在线播放| 毛片成人永久免费视频| 日本特黄特色aa大片免费| 欧美a级成人淫片免费看| 亚飞与亚基在线观看| 日韩欧美一及在线播放| 免费一级片在线观看| 日韩专区一区| 国产伦理精品| 欧美a级成人淫片免费看| 中文字幕Aⅴ资源网| 国产91视频网| 国产伦久视频免费观看视频| 日韩一级黄色| 久久国产影院| 国产不卡高清在线观看视频| 亚欧成人乱码一区二区| 黄色福利片| 成人高清免费| 沈樵在线观看福利| 韩国三级视频网站| 四虎影视库国产精品一区| 国产一区二区精品尤物| a级黄色毛片免费播放视频| 欧美另类videosbestsex高清| 亚洲 男人 天堂| 成人高清护士在线播放| 精品视频在线观看免费| 中文字幕Aⅴ资源网| 四虎影视久久| 久久福利影视| 欧美日本免费| 日本乱中文字幕系列| 国产一区精品| 成人高清视频在线观看| 黄色短视频网站| 国产福利免费观看| 一级毛片视频免费| 二级片在线观看| 91麻豆国产级在线| 一级女性大黄生活片免费| 99久久网站| 成人在免费观看视频国产| 成人高清视频在线观看| a级毛片免费观看网站| 99色视频在线| 九九精品在线| 91麻豆爱豆果冻天美星空| 一级女性大黄生活片免费| 成人影院一区二区三区| 黄色免费网站在线| 欧美大片aaaa一级毛片| a级精品九九九大片免费看| 你懂的国产精品| 韩国妈妈的朋友在线播放| 亚飞与亚基在线观看| 国产视频一区二区在线观看| 欧美大片一区| 午夜精品国产自在现线拍| 欧美日本韩国| 国产成人欧美一区二区三区的| 亚洲精品影院| 亚欧乱色一区二区三区| 你懂的福利视频| 四虎论坛| 美女免费毛片| 成人高清视频免费观看| 欧美激情一区二区三区视频| 国产麻豆精品高清在线播放| 欧美一级视频免费观看|