譯者:bat67可以使用最新版會在譯者倉庫首先同步。
torch.nn
包來構建神經網路.
我們以及介紹了autograd
,nn
包依賴於autograd
包來定義模型並對它們求導。乙個nn.module
包含各個層和乙個forward(input)
方法,該方法返回output
。
例如,下面這個神經網路可以對數字進行分類:
這是乙個簡單的前饋神經網路(feed-forward network)。它接受乙個輸入,然後將它送入下一層,一層接一層的傳遞,最後給出輸出。
乙個神經網路的典型訓練過程如下:
讓我們定義這樣乙個網路:
import torch
import torch.nn as nn
import torch.nn.functional as f
閱讀全文/改進本文
PyTorch 1 0 中文官方教程 遷移學習教程
譯者 片刻 作者 sasank chilamkurthy 在本教程中,您將學習如何使用遷移學習來訓練您的網路。您可以在 cs231n 筆記 上閱讀更多關於遷移學習的資訊 引用這些筆記 在實踐中,很少有人從頭開始訓練整個卷積網路 隨機初始化 因為擁有足夠大小的資料集是相對罕見的。相反,通常在非常大的資...
PyTorch 1 0 中文文件 資料型別資訊
譯者 馮寶寶 可以通過torch.finfo或torch.iinfo訪問torch.dtype的數字屬性。class torch finfotorch.finfo是乙個用來表示浮點torch.dtype的數字屬性的物件 即torch.float32,torch.float64和torch.float...
PyTorch 1 0宣布用於研究和生產AI專案
在最近的一篇博文中,臉書的副總裁bill jia宣布了pytorch的新版本1.0的發布。pytorch是用於python的開源ai框架軟體包,具有tensor計算,包括強大的gpu加速和建立於基於磁帶autodiff系統之上的深度神經網路。該新版本的重要性在於,減少了從研究過程到生產過程中移動ai...