fix headers for training apis (#14350)

Commit

3 years ago

fix headers for training apis (#14350) ### Description Minor refactor PR for fixing header placement for training apis