![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
. 小王同学.
这个作者很懒,什么都没留下…
展开
-
模型训练时GPU利用率太低的原因
最近在训练SSD模型时发现GPU的利用率只有8%,而CPU的利用率却非常高。后来了解到,一般使用CPU进行数据的读取和预处理,而使用GPU进行模型的正向传播和反向传播。由于CPU数据读取跟不上(读到内存+多线程+二进制文件),而GPU的处理速度太快,导致GPU的利用率不高。问题的解决办法:1,关闭一些日志记录;2,NVIDA提供了DALI库,可以将数据处理转移到GPU上。...原创 2020-08-23 18:23:23 · 14750 阅读 · 2 评论 -
VGG-16网络模型
前言最近在学习SSD,发现SSD采用VGG16作为基础模型,然后在VGG16的基础上新增了卷积层来获得更多的特征图用于检测。因此,这里简要介绍一下VGG16模型的结构。VGG16结构顾名思义,vgg16总共有16层,其中包括13个卷积层和3个全连接层。第一次经过64个卷积核的两次卷积后,采用一次pooling,第二次经过两次128个卷积核卷积后,再采用pooling,第三次经过两次256个卷积核卷积后,采用一次pooling,接着再重复两次三个512个卷积核卷积后,再pooling,最后经过三次全连接原创 2020-08-21 15:26:05 · 3192 阅读 · 0 评论