GPU memory usage - Githubissues

wywu / LAB

[CVPR 2018] Look at Boundary: A Boundary-Aware Face Alignment Algorithm

https://wywu.github.io/projects/LAB/LAB.html

Other

1.01k stars 273 forks source link

Open eloi-loomai opened 5 years ago

eloi-loomai commented 5 years ago

Hi, I'd like to know if there is something I can do to use this code with ~12GB of GPU memory

wywu commented 5 years ago

Hi, if you want to train it, you can try to use batch-size as 2 or 1.