Closed infrontofme closed 4 years ago
I solved it by adding del loss, output
after loss.backward
as well as reducing batch size
This method is not useful. Every time 20 cycles, out of memory appears. Is there any other way?thanks
Thanks a lot, it is solved.
I am reproducing the result using the instruction provided in the README file.
I am training base model with 1 GeForce GTX 1080 Ti with 12GB of memory. I modify batch_size=32.
when it runs about 20 epoches, cuda run time error occurs.
How can I solve this problem. Thanks:)