人工智能學習Pytorch數(shù)據(jù)集分割及動量示例詳解
更新時間：2021年11月11日 15:28:32 作者：Swayzzu
這篇文章主要為大家介紹了人工智能學習Pytorch數(shù)據(jù)集分割及動量示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步
1.數(shù)據(jù)集分割
通過datasets可以直接分別獲取訓練集和測試集。
通常我們會將訓練集進行分割，通過torch.utils.data.random_split方法。
所有的數(shù)據(jù)都需要通過torch.util.data.DataLoader進行加載，才可以得到可以使用的數(shù)據(jù)集。
具體代碼如下：
2.
2.正則化
PyTorch中的正則化和機器學習中的一樣，不過設置方式不一樣。
直接在優(yōu)化器中，設置weight_decay即可。優(yōu)化器中，默認的是L2范式，因此填入的參數(shù)就是lambda。想要使用L1范式的話，需要手動寫出代碼。
3.動量和學習率衰減
動量的設置可以直接在優(yōu)化器中完成。通過momentum參數(shù)設置。
學習率的調整通過torch.optim.lr_scheduler中的ReduceLROnPlateau,StepLR實現(xiàn)。
ReduceLROnPlateau是自動檢測損失值，并衰減學習率。
StepLR需要手動設置衰減的參數(shù)。
以上就是人工智能學習Pytorch數(shù)據(jù)集分割及動量示例詳解的詳細內容，更多關于pytorch數(shù)據(jù)集分割及動量的資料請關注腳本之家其它相關文章！
相關文章
這篇文章主要為大家詳細介紹了python實現(xiàn)簡單猜單詞游戲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2020-12-12
解決Pycharm無法import自己安裝的第三方module問題
今天小編就為大家分享一篇解決Pycharm無法import自己安裝的第三方module問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
這篇文章主要介紹了Python設計模式編程中的訪問者與觀察者模式,設計模式的制定有利于團隊協(xié)作編程代碼的協(xié)調,需要的朋友可以參考下2016-01-01
Python基于Opencv來快速實現(xiàn)人臉識別過程詳解（完整版）
這篇文章主要介紹了Python基于Opencv來快速實現(xiàn)人臉識別過程詳解（完整版）隨著人工智能的日益火熱，計算機視覺領域發(fā)展迅速，今天就為大家?guī)碜罨A的人臉識別基礎，從一個個函數(shù)開始走進這個奧妙的世界,需要的朋友可以參考下2019-07-07
reduce()函數(shù)即為化簡函數(shù)，它的執(zhí)行過程為：每一次迭代，都將上一次的迭代結果,需要的朋友可以參考下2017-11-11
詳解python 條件語句和while循環(huán)的實例代碼
這篇文章主要介紹了詳解python 條件語句和while循環(huán)，本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-12-12
這篇文章主要介紹了Python求解排列中的逆序數(shù)個數(shù)實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要介紹了使用Flask和Django中解決跨域請求問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
本文主要介紹了OpenCV仿射變換的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-08-08
使用python讀取.text文件特定行的數(shù)據(jù)方法
今天小編就為大家分享一篇使用python讀取.text文件特定行的數(shù)據(jù)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01