Skip to content

Eval_hooks中dataset.pre_eval()内存释放不正确的问题 #1071

@Syencil

Description

@Syencil

当eval数量比较多的时候,5000张左右,在训练一定程度后如果用pre_eval模式,CPU的物理内存会持续飙升到几十个G。怀疑是内存没有正常释放,可以持续复现。可以有1w张的数据做evaluation,检测机器的物理内存。
使用efficient_test模式内存则正常(应该是用了磁盘空间),测试速度约为pre_eval的65%左右,但是对内存十分友好(稳定在4G不到)
efficient_test这个参数在mmseg/core/evalution/eval_hooks.py 第50行左右 并没有作为参数传递到single_gpu_test函数中,会导致Config中的参数设定不可用

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions