bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码,可以用于训练自己的模型。
MIT License
787 stars 184 forks source link

为什么从rtx2060换成了双rtx3090训练速度没有提升呢 #48

Open wawaka1998 opened 3 years ago

wawaka1998 commented 3 years ago

我大幅度提高了batch_size,从之前的1直接提高到128,然而,我的两个3090半死不活的只用了6gb显存,cpu占用率倒是明显高了不少。而且每个batch_size训练的时间也低了很多倍,换算下来,训练速度并没有提高多少。

bubbliiiing commented 3 years ago

数据读取速度太慢了呢,你有用固态吗,多线程处理等,而且,tiny都不需要啥显卡呀。