As I recall, you can retrain units in Mongol Invasion if you've built the drill dojo in that province by dragging their unit icon up from the center panel to the training queue. That wasn't a feature in original STW because there was no drill dojo.