GPU memory usage

Open eloi-loomai opened this issue 7 years ago • 1 comments

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

Oct 20 '18 00:10 eloi-loomai

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

Oct 31 '18 12:10 wywu