search

jiyanggao / Video-Person-ReID

Video-based Person ReID Method Implementations on MARS
381 stars 110 forks source link

GPU Out of Memory #22

Closed DNHanXu closed 5 years ago

DNHanXu commented 5 years ago

I made some changes on original codes to make it compatible with torch 1.0.1 and python 3.6. However, I encounter gpu out of memory problem during testing process. Does anyone have the same problem? And is it caused by version difference? I used 1080ti with 11GB memory.

DNHanXu commented 5 years ago

make some changes according to version update of Pytorch from 0.3 to 0.4 will solve this problem

usename6 commented 5 years ago

I had same problem,could you tell me how to solve it?

DNHanXu commented 5 years ago

@usename6 For torch >= 0.4, try to use 'torch.no_grad()' instead of 'volatile' flag.