torch.no_grad()與.forward(）在預測時的性能差異

背景 使用pytorch在模型做推斷時，使用torch.no_grad()與.forward()都可以得到正確的預測結果，如下： 我好奇想知道這兩種推斷方式那種，那種效率更高，於是隨手做個測試。 測試 輸入一張圖片，然後推斷10000次，看下兩種方式各用多少時間： torch.no_grad(): .forward: 結論 我原本以爲這兩種方式推斷效率應該是差不多的，但測試結果告訴我，使用torc

>>阅读原文<<