search

hujiecpp / ISTR

ISTR: End-to-End Instance Segmentation with Transformers (https://arxiv.org/abs/2105.00637)
203 stars 28 forks source link

RuntimeError: CUDA out of memory. #18

Open maamouunn opened 2 years ago

maamouunn commented 2 years ago

return torch._C._nn.upsample_nearest2d(input, output_size, scale_factors) RuntimeError: CUDA out of memory. Tried to allocate 180.00 MiB (GPU 0; 10.92 GiB total capacity; 9.74 GiB already allocated; 57.56 MiB free; 9.90 GiB reserved in total by PyTorch)

how can i solve this issue? which file should i change ?

guangxuwang commented 2 years ago

how do you solve it？，my gpu is rtx 2080ti（memory 11G）.