[Error] MMdetection, CUDA error

[ed-kyu]

MMdetection, faster_rcnn_inference파일에서 (baseline 말고 깃헙 코드) https://github.com/boostcampaitech3/level2-object-detection-level2-cv-18/blob/master/Mmdetection/faster_rcnn_inference.ipynb

#checkpoint path
checkpoint_path = os.path.join(cfg.work_dir, f'{epoch}.pth')

model = build_detector(cfg.model, test_cfg=cfg.get('test_cfg')) # build detector
checkpoint = load_checkpoint(model, checkpoint_path, map_location='cpu') # ckpt load

model.CLASSES = dataset.CLASSES
model = MMDataParallel(model.cuda(), device_ids=[0]) # --------> CUDA error: out of memory

위 부분에서 오류가 나는데, 해결방법 아시는 분 있을까요

[ed-kyu]

faster_rcnn_train 파일 커널 죽이니까 되네요ㅎ..