A mistake on my side, though may be relevant to somebody. I overwrote the class TrainingArguments to add my own one, and the file with the new class (training_arguments.py in my case) must be present in the same directory from which you are uploading the arguments. Closing this.