cuda out of memory

[curry8]

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

[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