Spaces:
Runtime error
Runtime error
José Ángel González
commited on
Commit
·
e2092ff
1
Parent(s):
9513832
move to cpu and roberta-base alignscore
Browse files- generation_evaluator.py +4 -4
generation_evaluator.py
CHANGED
|
@@ -120,7 +120,7 @@ BERT_SCORE:{
|
|
| 120 |
"hashcode": Hashcode of the library.
|
| 121 |
},
|
| 122 |
AlignScore:{
|
| 123 |
-
"score": mean align scores using roberta-
|
| 124 |
},
|
| 125 |
CHRF:{
|
| 126 |
'score' (float): The chrF (chrF++) score,
|
|
@@ -131,10 +131,10 @@ CHRF:{
|
|
| 131 |
"""
|
| 132 |
|
| 133 |
ALIGNSCORE_ARGS = {
|
| 134 |
-
"model": "roberta-
|
| 135 |
"batch_size": 32,
|
| 136 |
-
"device": "
|
| 137 |
-
"ckpt_path": "https://huggingface.co/yzha/AlignScore/resolve/main/AlignScore-
|
| 138 |
"evaluation_mode": "nli_sp",
|
| 139 |
}
|
| 140 |
|
|
|
|
| 120 |
"hashcode": Hashcode of the library.
|
| 121 |
},
|
| 122 |
AlignScore:{
|
| 123 |
+
"score": mean align scores using roberta-base as scorer
|
| 124 |
},
|
| 125 |
CHRF:{
|
| 126 |
'score' (float): The chrF (chrF++) score,
|
|
|
|
| 131 |
"""
|
| 132 |
|
| 133 |
ALIGNSCORE_ARGS = {
|
| 134 |
+
"model": "roberta-base",
|
| 135 |
"batch_size": 32,
|
| 136 |
+
"device": "cpu",
|
| 137 |
+
"ckpt_path": "https://huggingface.co/yzha/AlignScore/resolve/main/AlignScore-base.ckpt",
|
| 138 |
"evaluation_mode": "nli_sp",
|
| 139 |
}
|
| 140 |
|