jogonba2 commited on
Commit
e2092ff
·
1 Parent(s): 9513832

move to cpu and roberta-base alignscore

Browse files
Files changed (1) hide show
  1. 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-large as scorer
124
  },
125
  CHRF:{
126
  'score' (float): The chrF (chrF++) score,
@@ -131,10 +131,10 @@ CHRF:{
131
  """
132
 
133
  ALIGNSCORE_ARGS = {
134
- "model": "roberta-large",
135
  "batch_size": 32,
136
- "device": "cuda",
137
- "ckpt_path": "https://huggingface.co/yzha/AlignScore/resolve/main/AlignScore-large.ckpt",
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