mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-02 11:11:05 +06:00
fix(integrations): consider test metrics (#13888)
This commit is contained in:
parent
0ddadbf0a8
commit
f099249cf1
@ -358,9 +358,13 @@ def rewrite_logs(d):
|
||||
new_d = {}
|
||||
eval_prefix = "eval_"
|
||||
eval_prefix_len = len(eval_prefix)
|
||||
test_prefix = "test_"
|
||||
test_prefix_len = len(test_prefix)
|
||||
for k, v in d.items():
|
||||
if k.startswith(eval_prefix):
|
||||
new_d["eval/" + k[eval_prefix_len:]] = v
|
||||
elif k.startswith(test_prefix):
|
||||
new_d["test/" + k[test_prefix_len:]] = v
|
||||
else:
|
||||
new_d["train/" + k] = v
|
||||
return new_d
|
||||
|
Loading…
Reference in New Issue
Block a user