Explicitly set number of training steps using Trainer

The number of steps is the number of update steps. It’s not the number of training examples to be seen.