I should point out the particulars I am using the pegasus-dailymail-cnn model. I have a small dataset of 14372, 819, 819 of train test val split. And currently my parameters are set to following and it is still causing me OOM errors. I will check out the given link. But I would really welcome any quick idea you can give me based on my current parameters
