search

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
6.94k stars 1.33k forks source link

CUDA error Out of memory #166

Closed minyoung-mia-Kim closed 4 years ago

minyoung-mia-Kim commented 4 years ago

Thank you for the great work first! But is there anyone who struggles Out of memory issues to get 3D landmark? I'm using three Geforce TitanX 12Gb in a server and allocating networks in different GPUs respectively. However, there is still a memory error. Please let me know how you all handle this!

Thank you for in advanced

charlescatta commented 4 years ago

Downsize your input image !

ex:

from skimage.transform import resize

img = resize(img, (img.shape[0] // 4, img.shape[1] // 4), anti_aliasing=True)