I am trying to use the same code (https://github.com/hashbangCoder/Text-Summarization) but I am still facing a CUDA OOM error after some iterations using the default settings? How did you manage to resolve this to get the model to train?