PyTorch_神經網絡詳解

神經網絡可使用torch.nn包。nn.Module包含層和方法,forward(input)返回output。 神經網絡的典型訓練過程以下: ------定義具備一些可學習參數(或權重)的神經網絡。 ------迭代輸入數據集 ------經過網絡處理輸入 ------計算損失(輸出離正確有多遠) ------將梯度傳播回網絡參數 ------更新網絡的權重,一般使用簡單的更新規則:weight
相關文章
相關標籤/搜索