Spaces:
Sleeping
Sleeping
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 |
|