Python深度學(xué)習(xí)pytorch神經(jīng)網(wǎng)絡(luò)多層感知機簡潔實現(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)
訓(xùn)練過程的實現(xiàn)與我們實現(xiàn)softmax回歸時完全相同,這種模塊化設(shè)計使我們能夠?qū)⑴c和模型架構(gòu)有關(guān)的內(nèi)容獨立出來。
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ò)多層感知機簡潔實現(xiàn)的詳細(xì)內(nèi)容,更多關(guān)于pytorch神經(jīng)網(wǎng)絡(luò)的資料請關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
Python+OpenCV實現(xiàn)單個圓形孔和針檢測
這篇文章主要為大家詳細(xì)介紹了如何通過Python+OpenCV實現(xiàn)單個圓形孔和針檢測功能,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解一下2022-10-10python實現(xiàn)把兩個二維array疊加成三維array示例
今天小編就為大家分享一篇python實現(xiàn)把兩個二維array疊加成三維array示例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11Python結(jié)合MySQL數(shù)據(jù)庫編寫簡單信息管理系統(tǒng)完整實例
最近Python課堂上布置了綜合實訓(xùn),實驗?zāi)繕?biāo)是設(shè)計一個信息管理系統(tǒng),下面這篇文章主要給大家介紹了關(guān)于Python結(jié)合MySQL數(shù)據(jù)庫編寫簡單信息管理系統(tǒng)的相關(guān)資料,文中通過實例代碼介紹的非常詳細(xì),需要的朋友可以參考下2023-06-06Python3實現(xiàn)Web網(wǎng)頁圖片下載
這篇文章主要介紹了Python3通過request.urlopen實現(xiàn)Web網(wǎng)頁圖片下載,感興趣的小伙伴們可以參考一下2016-01-01Python實現(xiàn)多并發(fā)訪問網(wǎng)站功能示例
這篇文章主要介紹了Python實現(xiàn)多并發(fā)訪問網(wǎng)站功能,結(jié)合具體實例形式分析了Python線程結(jié)合URL模塊并發(fā)訪問網(wǎng)站的相關(guān)操作技巧,需要的朋友可以參考下2017-06-06