debug
文章平均质量分 68
Cleo_Gao
这个作者很懒,什么都没留下…
展开
-
pytorch 训练过程内存泄露/显存泄露debug记录:dataloader和dataset导致的泄露
查找内存泄露/显存泄露的位置:把数据送入模型的代码全部注释掉,观察显存是否上涨;上涨说明内存泄露出现在dataloader(出现在非 dataloader 地方的最常见的显存泄露原因是,loss打印/统计的时候没有写把不同的 data 组成部分注释掉,观察具体是哪个 data 导致的内存泄露pytorch 释放内存的方法:把 tensor 读到 gpu 就会有显存占用,一般可以自动释放,但是显存泄露的时候就没法释放。找到没有及时释放的代码位置,然后首先del tensor标记删除,随后需要调用。原创 2023-08-06 09:55:06 · 4799 阅读 · 1 评论 -
debug: NameError: name ‘_C‘ is not defined 本地运行 GroundingDINO 代码 debug 记录
本地跑 Grounding DINO 过程记录原创 2023-05-08 17:18:03 · 4931 阅读 · 19 评论