transformers

refactor _inner_training_loop to smaller methods

#44041

Merged