Deep Learning
sean_xyz
问渠那得清如许,为有源头活水来
展开
-
深度学习工程实践Tips(长期更新)
1. 训练阶段加速 * 使用混合精度加速训练 推荐使用NVIDIA的APEX进行训练加速,在2060S显卡上,大约可以获得一倍的综合速度提升(显存使用减少,可以适当增加batchsize, 提高速度)。精度下降几乎察觉不到。APEX保存和加载的模型参数仍然是float32类型的,所以不用担心预训练模型重用的问题。使用起来也十分简单:h...原创 2019-12-13 11:09:22 · 414 阅读 · 0 评论 -
NMS极慢原因
近几天做了一个训练。以往都是使用迁移训练的方式,主干网络都是采用预训练的参数做初始化。但这次使用的是自己改进的网络。所以采用随机初始化开始训练。训练速度都还可以。但是在测试中间过程中,NMS(算法是Soft NMS)计算速度非常慢。(我的电脑2*1080Ti, 一般batch= 32, 单GPU的测试情况下,NMS只用0.008s)。而第一轮测试的时候,NMS计算居然到了0.18s,严重影响了...原创 2019-03-13 15:23:31 · 3415 阅读 · 2 评论 -
AI常用数据集下载
AI常用数据集下载:<持续更新>VOC 数据集下载:链接:https://pan.baidu.com/s/1zkqg62qKjj1rauAnvBUnXQ提取码:s6lhCamVid数据集下载:链接:https://pan.baidu.com/s/1W5zd1KBsiFltVd8f2ighDg提取码:spc7CityScape数据集下载:...原创 2019-07-30 10:54:17 · 2245 阅读 · 0 评论