Pytorch網絡-3

定義網絡 建一個class 定義網絡的層 定義網絡的前向傳播，反向傳播會自動推導。 Dataloader 2. 優化器&損失函數 訓練過程 model.train() 指明這個網絡有梯度，要更新參數。 optimizer.zero_grad() 優化器清零。 output = model(data) 計算前傳得到預測的輸出。 loss=F.nall_loss(output, target) 計算損

>>阅读原文<<