快捷導(dǎo)航

pytorch實(shí)現(xiàn)保證每次運(yùn)行使用的隨機(jī)數(shù)都相同

更新時(shí)間：2020年02月20日 11:17:33 作者：慢行厚積

今天小編就為大家分享一篇pytorch實(shí)現(xiàn)保證每次運(yùn)行使用的隨機(jī)數(shù)都相同，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

其實(shí)在代碼的開(kāi)頭添加下面幾句話即可：

# 保證訓(xùn)練時(shí)獲取的隨機(jī)數(shù)都是一樣的
init_seed = 1
torch.manual_seed(init_seed)
torch.cuda.manual_seed(init_seed)
np.random.seed(init_seed) # 用于numpy的隨機(jī)數(shù)

torch.manual_seed(seed)

為了生成隨機(jī)數(shù)設(shè)置種子。返回一個(gè)torch.Generator對(duì)象

參數(shù)：

seed (int) – 期望的種子數(shù)

torch.cuda.manual_seed(seed)

為當(dāng)前GPU生成隨機(jī)數(shù)設(shè)置種子。如果CUDA不可用，調(diào)用該方法也是安全的；在這種情況下，該調(diào)用就會(huì)被忽略

參數(shù)：

seed (int) – 期望的種子數(shù)

⚠️如果你使用的是多GPU模型，就要調(diào)用manual_seed_all(seed).

以上這篇pytorch實(shí)現(xiàn)保證每次運(yùn)行使用的隨機(jī)數(shù)都相同就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python腳本修改Maya ASCII文件路徑方法實(shí)現(xiàn)
本文主要介紹了Python腳本修改Maya ASCII文件路徑方法實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2023-02-02
python登錄并爬取淘寶信息代碼示例
這篇文章主要介紹了python 登錄并爬取淘寶信息代碼示例，具有一定借鑒價(jià)值，需要的朋友可以了解下。
2017-12-12
詳解Django+Uwsgi+Nginx的生產(chǎn)環(huán)境部署
這篇文章主要介紹了Django + Uwsgi + Nginx 的生產(chǎn)環(huán)境部署，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧
2018-06-06
Pyecharts繪制全球流向圖的示例代碼
這篇文章主要介紹了Pyecharts繪制全球流向圖的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2020-01-01
python 多維高斯分布數(shù)據(jù)生成方式
今天小編就為大家分享一篇python 多維高斯分布數(shù)據(jù)生成方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2019-12-12
Python中bisect的使用方法
這篇文章主要介紹了Python中bisect的使用方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2019-12-12
Python字典操作得力助手Get()函數(shù)的使用
在Python編程中，get()函數(shù)是字典（Dictionary）對(duì)象中非常有用的函數(shù)，本文將詳細(xì)介紹get()函數(shù)的用法及示例代碼，感興趣的可以了解一下
2023-11-11
最新Python?APScheduler?定時(shí)任務(wù)詳解
這篇文章主要介紹了Python使用apscheduler模塊設(shè)置定時(shí)任務(wù),APScheduler全稱(chēng)Advanced?Python?Scheduler?作用為在指定的時(shí)間規(guī)則執(zhí)行指定的作業(yè)，本文對(duì)Python?APScheduler?定時(shí)任務(wù)相關(guān)知識(shí)介紹的非常詳細(xì)，需要的朋友參考下
2022-05-05
python3使用libpcap庫(kù)進(jìn)行抓包及數(shù)據(jù)處理的操作方法
這篇文章主要介紹了python3使用libpcap庫(kù)進(jìn)行抓包及數(shù)據(jù)處理,需要的朋友可以參考下
2022-10-10
python實(shí)現(xiàn)不同數(shù)據(jù)庫(kù)間數(shù)據(jù)同步功能
這篇文章主要介紹了python實(shí)現(xiàn)不同數(shù)據(jù)庫(kù)間數(shù)據(jù)同步功能,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2021-02-02