Unverified Commit f38c4ad3 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

better logging and help (#9203)

parent e0e255be
...@@ -98,6 +98,7 @@ class DataTrainingArguments: ...@@ -98,6 +98,7 @@ class DataTrainingArguments:
metadata={ metadata={
"help": "The maximum total sequence length for validation target text after tokenization. Sequences longer " "help": "The maximum total sequence length for validation target text after tokenization. Sequences longer "
"than this will be truncated, sequences shorter will be padded." "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( test_max_target_length: Optional[int] = field(
......
...@@ -434,7 +434,8 @@ def use_task_specific_params(model, task): ...@@ -434,7 +434,8 @@ def use_task_specific_params(model, task):
if task_specific_params is not None: if task_specific_params is not None:
pars = task_specific_params.get(task, {}) 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) model.config.update(pars)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment