with torch.no_grad()

在講述with torch.no_grad()前,先從requires_grad講起

requires_grad

在pytorch中奸远,tensor有一個(gè)requires_grad參數(shù)磅氨,如果設(shè)置為True尺栖,則反向傳播時(shí),該tensor就會(huì)自動(dòng)求導(dǎo)烦租。tensor的requires_grad的屬性默認(rèn)為False,若一個(gè)節(jié)點(diǎn)(葉子變量:自己創(chuàng)建的tensor)requires_grad被設(shè)置為True延赌,那么所有依賴它的節(jié)點(diǎn)requires_grad都為True(即使其他相依賴的tensor的requires_grad = False)

x = torch.randn(10, 5, requires_grad = True)
y = torch.randn(10, 5, requires_grad = False)
z = torch.randn(10, 5, requires_grad = False)
w = x + y + z
w.requires_grad

True

volatile

首先說(shuō)明除盏,該用法已經(jīng)被移除,但為了說(shuō)明torch.no_grad挫以,還是需要講解下該作用者蠕。在之前的版本中,tensor(或者說(shuō)variable掐松,以前版本tensor會(huì)轉(zhuǎn)化成variable踱侣,目前該功能也被廢棄,直接使用tensor即可)還有一個(gè)參數(shù)volatile大磺,如果一個(gè)tensor的volatile = True抡句,那么所有依賴他的tensor會(huì)全部變成True,反向傳播時(shí)就不會(huì)自動(dòng)求導(dǎo)了杠愧,因此大大節(jié)約了顯存或者說(shuō)內(nèi)存待榔。
既然一個(gè)tensor既有requires_grad,又有volatile流济,那么當(dāng)兩個(gè)參數(shù)設(shè)置相矛盾時(shí)怎么辦锐锣?volatile=True的優(yōu)先級(jí)高于requires_grad,即當(dāng)volatile = True時(shí)绳瘟,無(wú)論requires_grad是Ture還是False雕憔,反向傳播時(shí)都不會(huì)自動(dòng)求導(dǎo)。volatile可以實(shí)現(xiàn)一定速度的提升稽荧,并節(jié)省一半的顯存橘茉,因?yàn)槠洳恍枰4嫣荻取#╲olatile默認(rèn)為False姨丈,這時(shí)反向傳播是否自動(dòng)求導(dǎo),取決于requires_grad)
with torch.no_grad
上文提到volatile已經(jīng)被廢棄擅腰,替代其功能的就是with torch.no_grad蟋恬。作用與volatile相似,即使一個(gè)tensor(命名為x)的requires_grad = True趁冈,由x得到的新tensor(命名為w-標(biāo)量)requires_grad也為False歼争,且grad_fn也為None,即不會(huì)對(duì)w求導(dǎo)。例子如下所示:

x = torch.randn(10, 5, requires_grad = True)
y = torch.randn(10, 5, requires_grad = True)
z = torch.randn(10, 5, requires_grad = True)
with torch.no_grad():
    w = x + y + z
    print(w.requires_grad)
    print(w.grad_fn)
print(w.requires_grad)


False
None
False

引用:https://blog.csdn.net/weixin_43178406/article/details/89517008

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末渗勘,一起剝皮案震驚了整個(gè)濱河市沐绒,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌旺坠,老刑警劉巖乔遮,帶你破解...
    沈念sama閱讀 219,490評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異取刃,居然都是意外死亡蹋肮,警方通過(guò)查閱死者的電腦和手機(jī)出刷,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,581評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)坯辩,“玉大人馁龟,你說(shuō)我怎么就攤上這事∑崮В” “怎么了坷檩?”我有些...
    開(kāi)封第一講書(shū)人閱讀 165,830評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)改抡。 經(jīng)常有香客問(wèn)我淌喻,道長(zhǎng),這世上最難降的妖魔是什么雀摘? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 58,957評(píng)論 1 295
  • 正文 為了忘掉前任裸删,我火速辦了婚禮,結(jié)果婚禮上阵赠,老公的妹妹穿的比我還像新娘涯塔。我一直安慰自己,他們只是感情好清蚀,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,974評(píng)論 6 393
  • 文/花漫 我一把揭開(kāi)白布匕荸。 她就那樣靜靜地躺著,像睡著了一般枷邪。 火紅的嫁衣襯著肌膚如雪榛搔。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 51,754評(píng)論 1 307
  • 那天东揣,我揣著相機(jī)與錄音践惑,去河邊找鬼。 笑死嘶卧,一個(gè)胖子當(dāng)著我的面吹牛尔觉,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播芥吟,決...
    沈念sama閱讀 40,464評(píng)論 3 420
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼侦铜,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了钟鸵?” 一聲冷哼從身側(cè)響起钉稍,我...
    開(kāi)封第一講書(shū)人閱讀 39,357評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎棺耍,沒(méi)想到半個(gè)月后贡未,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,847評(píng)論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,995評(píng)論 3 338
  • 正文 我和宋清朗相戀三年羞秤,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了缸托。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 40,137評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡瘾蛋,死狀恐怖俐镐,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情哺哼,我是刑警寧澤佩抹,帶...
    沈念sama閱讀 35,819評(píng)論 5 346
  • 正文 年R本政府宣布,位于F島的核電站取董,受9級(jí)特大地震影響棍苹,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜茵汰,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,482評(píng)論 3 331
  • 文/蒙蒙 一枢里、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧蹂午,春花似錦栏豺、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 32,023評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至晚胡,卻和暖如春灵奖,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背估盘。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 33,149評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工瓷患, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人忿檩。 一個(gè)月前我還...
    沈念sama閱讀 48,409評(píng)論 3 373
  • 正文 我出身青樓煌集,卻偏偏與公主長(zhǎng)得像露氮,于是被迫代替她去往敵國(guó)和親禾唁。 傳聞我的和親對(duì)象是個(gè)殘疾皇子庇绽,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,086評(píng)論 2 355