【pytorch】（deeplizard22-26）加載數據 | 訓練模型 | 計算梯度 | 優化

p22 將一個圖像傳遞到網絡中並輸出預測 理解前向傳播：將輸入張量轉換爲輸出張量的過程 將前面的一系列整合： 在開始之前，要關閉pytorch的梯度計算特性，這將阻止pytorch通過網絡將張量構建成一個計算圖。這個計算圖通過跟蹤張量在網絡中傳播的每一個計算來跟蹤網絡的映射，然後在訓練過程中使用這個圖形來計算導數（損失函數的導數），在訓練模型的時候纔打開。 預測的張量形狀是[1,10]，第一個軸長

>>阅读原文<<