PixelLink-with-pytorch icon indicating copy to clipboard operation
PixelLink-with-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon cheerss

cuda out of memory

Open curry8 opened this issue 6 years ago • 1 comments

when I train this model, it happened this situation. And I have a RTX 1080 Ti with 12GB memory.

curry8 avatar Nov 21 '19 03:11 curry8

ok,I just modified batch size to slove memory problem. But when I test my model, there is a bug of postprocess.py. link_neighbors[:, i] = link_neighbors[:, i] & pixel_class. I saw in other issues that you said this code will be updated, but it still exist. Looking forward to your reply

curry8 avatar Dec 09 '19 02:12 curry8