Move rouge loading
Merge pull request #2189 from benhanzl/move_rouge_loading