人工智能學(xué)習(xí)Pytorch數(shù)據(jù)集分割及動量示例詳解
1.數(shù)據(jù)集分割
通過datasets可以直接分別獲取訓(xùn)練集和測試集。
通常我們會將訓(xùn)練集進(jìn)行分割，通過torch.utils.data.random_split方法。
所有的數(shù)據(jù)都需要通過torch.util.data.DataLoader進(jìn)行加載，才可以得到可以使用的數(shù)據(jù)集。
具體代碼如下：
2.
2.正則化
PyTorch中的正則化和機(jī)器學(xué)習(xí)中的一樣，不過設(shè)置方式不一樣。
直接在優(yōu)化器中，設(shè)置weight_decay即可。優(yōu)化器中，默認(rèn)的是L2范式，因此填入的參數(shù)就是lambda。想要使用L1范式的話，需要手動寫出代碼。
3.動量和學(xué)習(xí)率衰減
動量的設(shè)置可以直接在優(yōu)化器中完成。通過momentum參數(shù)設(shè)置。
學(xué)習(xí)率的調(diào)整通過torch.optim.lr_scheduler中的ReduceLROnPlateau,StepLR實(shí)現(xiàn)。
ReduceLROnPlateau是自動檢測損失值，并衰減學(xué)習(xí)率。
StepLR需要手動設(shè)置衰減的參數(shù)。
以上就是人工智能學(xué)習(xí)Pytorch數(shù)據(jù)集分割及動量示例詳解的詳細(xì)內(nèi)容，更多關(guān)于pytorch數(shù)據(jù)集分割及動量的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
K-means算法是最為經(jīng)典的基于劃分的聚類方法,是十大經(jīng)典數(shù)據(jù)挖掘算法之一,下面這篇文章主要給大家介紹了關(guān)于Python自定義指標(biāo)聚類的相關(guān)資料,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-02-02
這篇文章主要給大家分享得是 python中的3種定義類方法，分別是普通方法、類方法(@classmethod)以及類方法(@classmethod)，想了解方法具體實(shí)現(xiàn)的小伙伴可以參考下面文章內(nèi)容哦，希望對你有所幫助2021-11-11
本文詳細(xì)講解了Python基礎(chǔ)之模塊，文中通過示例代碼介紹的非常詳細(xì)。對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-05-05
Python中re模塊主要功能是通過正則表達(dá)式是用來匹配處理字符串的 ,下面這篇文章主要給大家介紹了關(guān)于Python正則表達(dá)式re模塊講解以及其案例舉例的相關(guān)資料,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-09-09
Python模擬登陸網(wǎng)頁的三種方法小結(jié)
如何使用Python模擬登陸網(wǎng)頁,尤其是在涉及到復(fù)雜的認(rèn)證機(jī)制時？這篇文章將詳細(xì)介紹Python模擬登陸網(wǎng)頁的三種方法,以及如何繞過一些常見的安全防護(hù)措施,需要的朋友可以參考下2024-01-01
JSON Web Tokens的實(shí)現(xiàn)原理
本文主要介紹了JSON Web Tokens的實(shí)現(xiàn)原理。具有很好的參考價值，下面跟著小編一起來看下吧2017-04-04
這篇文章主要給大家介紹了關(guān)于Python打包時包含字庫文件的幾種常見方法,通過示例講解了基本方法、使用spec文件、批量添加字體文件夾和在代碼中訪問字體文件的注意事項(xiàng),需要的朋友可以參考下2025-05-05
python掃描proxy并獲取可用代理ip的實(shí)例
下面小編就為大家?guī)硪黄猵ython掃描proxy并獲取可用代理ip的實(shí)例。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-08-08
keras 實(shí)現(xiàn)輕量級網(wǎng)絡(luò)ShuffleNet教程
這篇文章主要介紹了keras 實(shí)現(xiàn)輕量級網(wǎng)絡(luò)ShuffleNet教程，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
這篇文章講給大家介紹一下python使用pyecharts繪制簡單的折線圖的黨法步驟，文中有詳細(xì)的代碼示例講解，對我們學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下2023-07-07