YOLOv5在COCO數(shù)據(jù)集上訓(xùn)練需要的內(nèi)存大小

YOLOv5在COCO數(shù)據(jù)集上訓(xùn)練需要的內(nèi)存大蟹柞恕:128GB
測(cè)試過程中,虛擬內(nèi)存使用了61GB邀杏,內(nèi)存使用了14.1GB贫奠,總數(shù)為75GB唬血,但在這種情況下,還報(bào)錯(cuò)唤崭!

內(nèi)存不足報(bào)錯(cuò)

實(shí)驗(yàn)數(shù)據(jù)

注意:

  1. 對(duì)于workers拷恨,并不是越大越好,太大時(shí)gpu其實(shí)處理不過來谢肾,訓(xùn)練速度一樣腕侄,但虛擬內(nèi)存(磁盤空間)會(huì)成倍占用;對(duì)于單機(jī)單卡芦疏,workers=4就好冕杠,大于4,訓(xùn)練速度也沒有明顯提升酸茴,反而增加了虛擬內(nèi)存的使用
  2. 對(duì)于batch-size分预,實(shí)際測(cè)試下來,發(fā)現(xiàn)當(dāng)為8的倍數(shù)時(shí)效率更高一點(diǎn)薪捍,即 batch-size 32的效率反而會(huì)高于batch-size 34笼痹。

實(shí)際測(cè)試下來,在32GB內(nèi)存酪穿,1080Ti 11G顯存的系統(tǒng)上凳干,workers=4,batch-size=32基本達(dá)到上限了被济。

workers=4纺座,batch-size=32

workers=4,batch-size=32配置下溉潭,內(nèi)存=10GB,虛擬內(nèi)存50GB少欺,總共需要60GB喳瓣,64GB內(nèi)存可能ok,但是要測(cè)試赞别!

內(nèi)存消耗60G

workers=4畏陕,batch-size=32配置下,經(jīng)過測(cè)試仿滔,內(nèi)存穩(wěn)定在11.8GB惠毁,虛擬內(nèi)存45.7/56.1GB,顯存訓(xùn)練時(shí)6.4GB崎页,評(píng)估時(shí)會(huì)增加到10.6GB鞠绰。證明在單機(jī)單卡上,workers=4飒焦,batch-size=32已經(jīng)基本榨干了1080Ti了

YOLOv5s在COCO數(shù)據(jù)集上訓(xùn)練

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末蜈膨,一起剝皮案震驚了整個(gè)濱河市屿笼,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌翁巍,老刑警劉巖驴一,帶你破解...
    沈念sama閱讀 217,542評(píng)論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異灶壶,居然都是意外死亡肝断,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,822評(píng)論 3 394
  • 文/潘曉璐 我一進(jìn)店門驰凛,熙熙樓的掌柜王于貴愁眉苦臉地迎上來胸懈,“玉大人,你說我怎么就攤上這事洒嗤◇锏矗” “怎么了?”我有些...
    開封第一講書人閱讀 163,912評(píng)論 0 354
  • 文/不壞的土叔 我叫張陵渔隶,是天一觀的道長(zhǎng)羔挡。 經(jīng)常有香客問我,道長(zhǎng)间唉,這世上最難降的妖魔是什么绞灼? 我笑而不...
    開封第一講書人閱讀 58,449評(píng)論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮呈野,結(jié)果婚禮上低矮,老公的妹妹穿的比我還像新娘。我一直安慰自己被冒,他們只是感情好军掂,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,500評(píng)論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著昨悼,像睡著了一般蝗锥。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上率触,一...
    開封第一講書人閱讀 51,370評(píng)論 1 302
  • 那天终议,我揣著相機(jī)與錄音,去河邊找鬼葱蝗。 笑死穴张,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的两曼。 我是一名探鬼主播皂甘,決...
    沈念sama閱讀 40,193評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼悼凑!你這毒婦竟也來了叮贩?” 一聲冷哼從身側(cè)響起击狮,我...
    開封第一講書人閱讀 39,074評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎益老,沒想到半個(gè)月后彪蓬,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,505評(píng)論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡捺萌,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,722評(píng)論 3 335
  • 正文 我和宋清朗相戀三年档冬,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片桃纯。...
    茶點(diǎn)故事閱讀 39,841評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡酷誓,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出态坦,到底是詐尸還是另有隱情盐数,我是刑警寧澤,帶...
    沈念sama閱讀 35,569評(píng)論 5 345
  • 正文 年R本政府宣布伞梯,位于F島的核電站玫氢,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏谜诫。R本人自食惡果不足惜漾峡,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,168評(píng)論 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望喻旷。 院中可真熱鬧生逸,春花似錦、人聲如沸且预。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,783評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)锋谐。三九已至掰伸,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間怀估,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,918評(píng)論 1 269
  • 我被黑心中介騙來泰國(guó)打工合搅, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留多搀,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 47,962評(píng)論 2 370
  • 正文 我出身青樓灾部,卻偏偏與公主長(zhǎng)得像康铭,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子赌髓,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,781評(píng)論 2 354

推薦閱讀更多精彩內(nèi)容