遇到這種情況 ——深度學(xué)習(xí)

首先檢查自己的數(shù)據(jù)下載部分，是不是出現(xiàn)重復(fù)下載。這里的重復(fù)下載是指在訓(xùn)練源數(shù)據(jù)的部分，單獨(dú)一個(gè)函數(shù)train(train_loader,target_loader)中存在數(shù)據(jù)ids與data下載，不要將其放置在真正的循環(huán)訓(xùn)練中（因?yàn)楸救藶榱巳菀桌斫?#xff0c;將其放入循環(huán)中了導(dǎo)致程序運(yùn)行很慢）；

其次檢查DataLoader中的num_workers=？，我也不清楚為什么這里設(shè)置為8，會(huì)使程序運(yùn)行加快。總之，將其內(nèi)部參數(shù)設(shè)置為pin_memory=True, num_workers=8。

我看到有的鏈接說(shuō)是因?yàn)槲募蘒O操作導(dǎo)致的，也就是log_interval的值需要適當(dāng)變大，這個(gè)我不太理解到底還需要在哪里需要修改。

目前我的程序比之前快了一些，但是還不是很快。

與50位技術(shù)專家面對(duì)面20年技術(shù)見(jiàn)證，附贈(zèng)技術(shù)全景圖

總結(jié)

以上是生活随笔為你收集整理的模型训练速度过慢，GPU利用率低的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。