Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
4c33a0c4
"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "52a46dc57bb653aa9dab440e4bb70988b15cdc7e"
Unverified
Commit
4c33a0c4
authored
Apr 03, 2023
by
Ilya
Committed by
GitHub
Apr 03, 2023
Browse files
Fix missing metrics with multiple eval datasets (#22536)
parent
d7a4f5be
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
src/transformers/trainer.py
src/transformers/trainer.py
+3
-1
No files found.
src/transformers/trainer.py
View file @
4c33a0c4
...
@@ -2243,12 +2243,14 @@ class Trainer:
...
@@ -2243,12 +2243,14 @@ class Trainer:
metrics
=
None
metrics
=
None
if
self
.
control
.
should_evaluate
:
if
self
.
control
.
should_evaluate
:
if
isinstance
(
self
.
eval_dataset
,
dict
):
if
isinstance
(
self
.
eval_dataset
,
dict
):
metrics
=
{}
for
eval_dataset_name
,
eval_dataset
in
self
.
eval_dataset
.
items
():
for
eval_dataset_name
,
eval_dataset
in
self
.
eval_dataset
.
items
():
metrics
=
self
.
evaluate
(
dataset_
metrics
=
self
.
evaluate
(
eval_dataset
=
eval_dataset
,
eval_dataset
=
eval_dataset
,
ignore_keys
=
ignore_keys_for_eval
,
ignore_keys
=
ignore_keys_for_eval
,
metric_key_prefix
=
f
"eval_
{
eval_dataset_name
}
"
,
metric_key_prefix
=
f
"eval_
{
eval_dataset_name
}
"
,
)
)
metrics
.
update
(
dataset_metrics
)
else
:
else
:
metrics
=
self
.
evaluate
(
ignore_keys
=
ignore_keys_for_eval
)
metrics
=
self
.
evaluate
(
ignore_keys
=
ignore_keys_for_eval
)
self
.
_report_to_hp_search
(
trial
,
self
.
state
.
global_step
,
metrics
)
self
.
_report_to_hp_search
(
trial
,
self
.
state
.
global_step
,
metrics
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment