memory leak problem when training

qianguih / voxelnet

This is an unofficial inplementation of VoxelNet in TensorFlow.

654 stars 241 forks source link

memory leak problem when training #60

Closed TonyXuQAQ closed 2 years ago

TonyXuQAQ commented 5 years ago

Does any met the memory leak problem when training? My PC memory (16GB) is full after around 600 steps training. After exploring the code, I found that this is caused by rets=TRAIN_POOL.map(proc,excerpt) in kitti_loader.py. Does anyone met the similar problem and has solutions to this ?