Implement fixes for TrainingArguments doc
Co-authored-by: osanseviero <osanseviero@gmail.com>

This implements the suggestions I didn't add in #15327 because I forgot to click on a button :-/ 
_The documentation is not available anymore as the PR was closed or merged._