search

lipiji / TranSummar

Transformer for abstractive summarization on cnn/daily-mail and gigawords
MIT License
140 stars 20 forks source link

老是报这个问题：RuntimeError: CUDA error: out of memory #7

Closed dajuzi123 closed 5 years ago

dajuzi123 commented 5 years ago

老是报这个问题：RuntimeError: CUDA error: out of memory 把哪些参数调一下可以让它跑通，我不懂的有点多，麻烦了

lipiji commented 5 years ago

try to decrease the batch size.