search

pmj110119 / RenderOcc

[ICRA 2024] RenderOcc: Vision-Centric 3D Occupancy Prediction with 2D Rendering Supervision. (Early version: UniOcc)
447 stars 25 forks source link

cuda memory #16

Closed Yzichen closed 1 year ago

Yzichen commented 1 year ago

Why does the cuda memory footprint keep changing during training.

pmj110119 commented 1 year ago

Sorry, I haven't encountered this problem before.

chenzyhust commented 1 year ago

Why does the cuda memory footprint keep changing during training. 注释掉lru_cache就可以了，不然服务器内存不够会爆掉

@functools.lru_cache(maxsize=32)

def create_full_step_id(shape):