Fix transformer API for training-evaluation pipeline#2018

jeffra merged 3 commits intomasterdeepspeedai/DeepSpeed:masterfrom

transformer-kernel/fix-train-evaldeepspeedai/DeepSpeed:transformer-kernel/fix-train-evalCopy head branch name to clipboard