PRSummarizer icon indicating copy to clipboard operation
PRSummarizer copied to clipboard

Published 20 hours ago • verified publisher icon Tbabm

Update prsum.py

Open TristanWH opened this issue 2 years ago • 0 comments

I and my colleagues got "RuntimeError:CUDA out of memory",withRTX2060Ti, add" with torch.no_grad():"means stop the back propagation and it works to us.

TristanWH avatar Nov 12 '22 19:11 TristanWH