Python深度學習pytorch神經網絡多層感知機簡潔實現(xiàn)

更新時間：2021年10月11日 15:47:11 作者：Supre_yuan

這篇文章主要為大家講解了Python深層學習中pytorch神經網絡多層感知機的簡潔實現(xiàn)方式，有需要的朋友可以借鑒參考下，希望能夠有所幫助

我們可以通過高級API更簡潔地實現(xiàn)多層感知機。

import torch
from torch import nn
from d2l import torch as d2l

模型

與softmax回歸的簡潔實現(xiàn)相比，唯一的區(qū)別是我們添加了2個全連接層。第一層是隱藏層，它包含256個隱藏單元，并使用了ReLU激活函數(shù)。第二層是輸出層。

net = nn.Sequential(nn.Flatten(),
					nn.Linear(784, 256),
					nn.ReLU(),
					nn.Linear(256, 10))
def init_weights(m):
	if type(m) == nn.Linear:
		nn.init.normal_(m.weight, std=0.01)

net.apply(init_weights)

訓練過程的實現(xiàn)與我們實現(xiàn)softmax回歸時完全相同，這種模塊化設計使我們能夠將與和模型架構有關的內容獨立出來。

batch_size, lr, num_epochs = 256, 0.1, 10    # 批量大小為256，學習率為0.1，類型為10
loss = nn.CrossEntropyLoss()    # 使用交叉熵損失函數(shù)
trainer = torch.optim.SGD(net.parameters(), lr=lr)    # 開始訓練

train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size)
d2l.train_ch3(net, train_iter, test_iter, loss, num_epochs, trainer)

在這里插入圖片描述

以上就是Python深度學習pytorch神經網絡多層感知機簡潔實現(xiàn)的詳細內容，更多關于pytorch神經網絡的資料請關注腳本之家其它相關文章！

您可能感興趣的文章:

詳細介紹Python中的偏函數(shù)
這篇文章主要介紹了Python中的偏函數(shù),示例代碼基于Python2.x版本,需要的朋友可以參考下
2015-04-04
pytorch實現(xiàn)模型剪枝的操作方法
PyTorch 提供了內置剪枝 API ，也支持了一些非結構化和結構化剪枝方法，但是 API 比較混亂，對應文檔描述也不清晰，所以后面我還會結合微軟的開源 nni 工具來實現(xiàn)模型剪枝功能,這篇文章主要介紹了pytorch實現(xiàn)模型剪枝, 需要的朋友可以參考下
2023-02-02
Python+OpenCV實現(xiàn)單個圓形孔和針檢測
這篇文章主要為大家詳細介紹了如何通過Python+OpenCV實現(xiàn)單個圓形孔和針檢測功能，文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下
2022-10-10
python實現(xiàn)把兩個二維array疊加成三維array示例
今天小編就為大家分享一篇python實現(xiàn)把兩個二維array疊加成三維array示例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-11-11
Python中如何生成GeoJSON數(shù)據
這篇文章主要介紹了Python中生成GeoJSON數(shù)據,無論使用geojson庫還是geopandas庫，都可以生成包含地理空間數(shù)據的GeoJSON文件，文中介紹了使用這些庫生成GeoJSON數(shù)據的簡單示例，需要的朋友可以參考下
2023-10-10
Python結合MySQL數(shù)據庫編寫簡單信息管理系統(tǒng)完整實例
最近Python課堂上布置了綜合實訓,實驗目標是設計一個信息管理系統(tǒng),下面這篇文章主要給大家介紹了關于Python結合MySQL數(shù)據庫編寫簡單信息管理系統(tǒng)的相關資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下
2023-06-06
Python3實現(xiàn)Web網頁圖片下載
這篇文章主要介紹了Python3通過request.urlopen實現(xiàn)Web網頁圖片下載，感興趣的小伙伴們可以參考一下
2016-01-01
Python PyInstaller庫基本使用方法分析
這篇文章主要介紹了Python PyInstaller庫基本使用方法,結合實例形式分析了Python PyInstaller庫的功能、安裝及相關使用注意事項,需要的朋友可以參考下
2019-12-12
Python實現(xiàn)多并發(fā)訪問網站功能示例
這篇文章主要介紹了Python實現(xiàn)多并發(fā)訪問網站功能,結合具體實例形式分析了Python線程結合URL模塊并發(fā)訪問網站的相關操作技巧,需要的朋友可以參考下
2017-06-06
python xpath獲取頁面注釋的方法
今天小編就為大家分享一篇python xpath獲取頁面注釋的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-01-01