I just run the code for depth completion task :  https://github.com/abdo-eldesokey/nconv 6
(I also open an issue in that repository…)
I try to only load data and do nothing for test like:

image.png885×328 15.6 KB

the cached mem still increases…
The dataset I used is KITTI dataset for depth completion 1, with resolution 1242x375 (center crop to 1216x352), batch_size=8, and I try to set the dataloader num_worker=0, it’s still the same situation.
Thank you!