CUDA out of memory when optimizer.step()

[shirui-japina (Shirui Zhang)]

I got it. Thanks for your advising