PyTorch 從 v1.9 開始具有新功能torch.inference_mode,它“類似于 torch.no_grad……通過禁用視圖跟蹤和版本計數器顛簸,在此模式下運行的代碼可以獲得更好的性能。”
如果我只是在測驗時評估我的模型(即不是訓練),有沒有比torch.no_grad更可取的情況torch.inference_mode?我計劃用后者替換前者的每個實體,并且我希望使用運行時錯誤作為護欄(即我相信任何問題都會顯示為運行時錯誤,如果它沒有作為運行時錯誤出現,那么我認為使用torch.inference_mode)確實更可取。
PyTorch 開發人員播客中提到了有關為什么開發推理模式的更多詳細資訊。
uj5u.com熱心網友回復:
是的,torch.inference_mode是真的最好要torch.no_grad在推理模式不會拋出運行時錯誤的所有情況。
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/317863.html
