BUG!!!!!GPU MEMORY LEAK MAY EXIST!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

mli0603 / stereo-transformer

Revisiting Stereo Depth Estimation From a Sequence-to-Sequence Perspective with Transformers. (ICCV 2021 Oral)

Apache License 2.0

664 stars 106 forks source link

BUG!!!!!GPU MEMORY LEAK MAY EXIST!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! #70

Closed sunyuhan19981208 closed 2 years ago

sunyuhan19981208 commented 2 years ago

CUDA OOM exception was raised in tier 90/374, and this 12GB 3080TI GPU is only used by myself. Therefore it's very possible that memory leak exists.

sunyuhan19981208 commented 2 years ago

I tune up the downsample option to 4, and it does not raise exception anymore, so I guess maybe the 90th picture is bigger than others, and it caused the OOM, MEMORY LEAK may not exist.