Xt-Chen / SARPN

Structure-Aware Residual Pyramid Network for Monocular Depth Estimation IJCAI 2019
86 stars 15 forks source link

RuntimeError: CUDA out of memory. #12

Open abbaskali opened 2 years ago

abbaskali commented 2 years ago

Dear @Xt-Chen, i got this problem how i can solve it, by the way i decreased Batch_size until 1 but does not solve give me an Error on this line: loading model ./checkpoints/SARPN_checkpoints_20.pth.tar Processing the 0th image! Traceback (most recent call last): File "evaluate.py", line 95, in test() File "evaluate.py", line 52, in test pred_depth = model(image)

RuntimeError: CUDA out of memory. Tried to allocate 22.00 MiB (GPU 0; 8.00 GiB total capacity; 1.27 GiB already allocated; 0 bytes free; 1.30 GiB reserved in total by PyTorch)