實戰人工智能框架PyTorch的模型訓練

咱們關心網絡結構和數據,定義損失函數,定義優化函數等。算法 具體步驟以下:網絡 第一步:圖像封裝爲向量後,將輸入input向前傳播,進行運算後獲得輸出outputdom 第二步:將output再輸入loss函數,計算loss值(是個標量)-損失函數用來獲得新權重函數 第三步:將梯度反向傳播到每一個參數(優化函數):主要指標是 學習速率𝜂  x 梯度向量g學習 第四步:利用下面公式進行權重更新優化
相關文章
相關標籤/搜索