modify training corpus loading