快捷導(dǎo)航

Python深度學(xué)習(xí)pytorch神經(jīng)網(wǎng)絡(luò)多層感知機(jī)簡潔實(shí)現(xiàn)

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

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

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

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

模型

與softmax回歸的簡潔實(shí)現(xiàn)相比，唯一的區(qū)別是我們添加了2個(gè)全連接層。第一層是隱藏層，它包含256個(gè)隱藏單元，并使用了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)

訓(xùn)練過程的實(shí)現(xiàn)與我們實(shí)現(xiàn)softmax回歸時(shí)完全相同，這種模塊化設(shè)計(jì)使我們能夠?qū)⑴c和模型架構(gòu)有關(guān)的內(nèi)容獨(dú)立出來。

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

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

在這里插入圖片描述

以上就是Python深度學(xué)習(xí)pytorch神經(jīng)網(wǎng)絡(luò)多層感知機(jī)簡潔實(shí)現(xiàn)的詳細(xì)內(nèi)容，更多關(guān)于pytorch神經(jīng)網(wǎng)絡(luò)的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章:

相關(guān)文章

Python Pandas工具繪制數(shù)據(jù)圖使用教程
Pandas是基于NumPy 的一種工具，該工具是為解決數(shù)據(jù)分析任務(wù)而創(chuàng)建的。Pandas還可以繪制多重?cái)?shù)據(jù)圖表，本文將為大家介紹如何通過Pandas繪制圖表，感興趣的可以了解一下
2021-12-12
python對數(shù)組進(jìn)行反轉(zhuǎn)的方法
這篇文章主要介紹了python對數(shù)組進(jìn)行反轉(zhuǎn)的方法,涉及Python中reverse方法的使用技巧,需要的朋友可以參考下
2015-05-05
Python連接PostgreSQL數(shù)據(jù)庫并查詢數(shù)據(jù)的詳細(xì)指南
在現(xiàn)代軟件開發(fā)中,數(shù)據(jù)庫是存儲(chǔ)和檢索數(shù)據(jù)的核心組件,PostgreSQ是一個(gè)功能強(qiáng)大的開源對象關(guān)系數(shù)據(jù)庫系統(tǒng),它以其穩(wěn)定性、強(qiáng)大的功能和靈活性而聞名,Python作為一種流行的編程語言,與PostgreSQL的結(jié)合使用非常廣泛,本文介紹了Python連接PostgreSQL數(shù)據(jù)庫并查詢數(shù)據(jù)
2024-12-12
Python繪制莖葉圖的示例代碼
這篇文章主要為大家信息介紹了Python繪制莖葉圖的相關(guān)知識(shí),文中的示例代碼講解詳細(xì),具有一定的借鑒價(jià)值,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下
2024-01-01
python openCV自制繪畫板
這篇文章主要為大家詳細(xì)介紹了python openCV自制繪畫板，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2020-10-10
Python常用數(shù)據(jù)分析模塊原理解析
這篇文章主要介紹了Python常用數(shù)據(jù)分析模塊原理解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-07-07
python的tkinter中如何改變label背景顏色
在Python的tkinter中,使用config方法的background屬性可以改變label的背景顏色
2024-11-11
Pycharm編輯器功能之代碼折疊效果的實(shí)現(xiàn)代碼
這篇文章主要介紹了Pycharm編輯器功能之代碼折疊效果的實(shí)現(xiàn)代碼,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2020-10-10
Python中八大圖像特效算法的示例詳解
圖像特效處理是基于圖像像素?cái)?shù)據(jù)特征，將原圖像進(jìn)行一定步驟的計(jì)算——例如像素作差、灰度變換、顏色通道融合等，從而達(dá)到期望的效果。本文將為大家介紹八種常用的圖像特效算法，需要的可以參考一下
2022-03-03
基于Python實(shí)現(xiàn)在線加密解密網(wǎng)站系統(tǒng)
在這個(gè)數(shù)字化時(shí)代,數(shù)據(jù)的安全和隱私變得越來越重要,所以本文小編就來帶大家實(shí)現(xiàn)一個(gè)簡單但功能強(qiáng)大的加密解密系統(tǒng),并深入探討它是如何工作的,有興趣的可以了解下
2023-09-09