model.config.use_cache = False

Define the training arguments.