better logging and help (#9203)

This commit is contained in:
Stas Bekman 2020-12-20 10:28:28 -08:00 committed by GitHub
parent e0e255be1f
commit f38c4ad302
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions

View File

@ -98,6 +98,7 @@ class DataTrainingArguments:
metadata={
"help": "The maximum total sequence length for validation target text after tokenization. Sequences longer "
"than this will be truncated, sequences shorter will be padded."
" This argument is also used to override the ``max_length`` param of ``model.generate``, which is used during ``evaluate`` and ``predict``"
},
)
test_max_target_length: Optional[int] = field(

View File

@ -434,7 +434,8 @@ def use_task_specific_params(model, task):
if task_specific_params is not None:
pars = task_specific_params.get(task, {})
logger.info(f"using task specific params for {task}: {pars}")
logger.info(f"setting model.config to task specific params for {task}:\n {pars}")
logger.info("note: command line args may override some of these")
model.config.update(pars)