深入解析OpenMMLab Hook机制：面向切面编程的设计理念与应用
OpenMMLab 系列的一大特色是其所采用的 Hook 机制。同样作为计算机视觉算法框架, 相比于简单易懂、对新手友好的 Gluon-CV...
qer1235天前13
pytorch 计算过程中如何节省显存并注册钩子导出中间变量
文章浏览阅读1.9w次，点赞53次，收藏130次。 为了节省显存（内存），pytorch在计算过程中不保存中间变量，包括中间层的特征图和非叶子张量的梯度等。...
qer1232个月前 (11-06)36
