溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

Pytorch中with torch.no_grad()方法如何使用

發(fā)布時間：2021-06-23 14:31:40 來源：億速云 閱讀：2173 作者：Leah 欄目：大數(shù)據(jù)

Pytorch中with torch.no_grad()方法如何使用，針對這個問題，這篇文章詳細(xì)介紹了相對應(yīng)的分析和解答，希望可以幫助更多想解決這個問題的小伙伴找到更簡單易行的方法。

  • requires_grad=True 要求計算梯度

  • requires_grad=False 不要求計算梯度

  • with torch.no_grad()或者@torch.no_grad()中的數(shù)據(jù)不需要計算梯度，也不會進行反向傳播

model.eval()                                # 測試模式
with torch.no_grad():
   pass
@torch.no_grad()
def eval():
	...

關(guān)于Pytorch中with torch.no_grad()方法如何使用問題的解答就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，如果你還有很多疑惑沒有解開，可以關(guān)注億速云行業(yè)資訊頻道了解更多相關(guān)知識。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手