摘要: 在进行多卡训练的时候,经常会出现GPU利用率上不来的情况,无法发挥硬件的最大实力。 造成这种现象最有可能的原因是,CPU生成数据的能力,已经跟不上GPU处理数据的能力。 方法一 常见的方法为修改Dataloader里面的线程数量,利用多线程技术提高数据生产能力,但是这种方法提速并不是特别明显。 tr 阅读全文
posted @ 2022-07-04 14:13 cwpeng 阅读(1363) 评论(0) 推荐(1) 编辑
摘要: 在众多深度学习开源库的代码中经常出现Registry代码块,例如OpenMMlab,facebookresearch和BasicSR中都使用了注册器机制。这块的代码经常会让新使用这些库的初学者感到一头雾水,本篇博客来分析一下注册器机制的原理与好处。 1. 为什么使用registry 在讲解regis 阅读全文
posted @ 2022-07-04 14:07 cwpeng 阅读(1121) 评论(0) 推荐(0) 编辑