Optimize the Trompt example to reduce training time by ~30%

Optimize the Trompt example to reduce training time by ~30% #856