MiriUll commited on
Commit
3ec7eda
·
verified ·
1 Parent(s): 739c9dd

Corrected checkpoint parameter

Browse files

The checkpoint parameter is never used and will be ignored. It should be set to `config_name`, so that the [EvaluationModule](https://github.com/huggingface/evaluate/blob/5aa3982a9a8c86e506860e381d428a64b0cce73b/src/evaluate/module.py#L149) reads it.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -44,7 +44,7 @@ This metric takes as input lists of predicted sentences and reference sentences:
44
  ### Inputs
45
  - **predictions** (`list` of `str`s): List of generated sentences to score.
46
  - **references** (`list` of `str`s): List of references to compare to.
47
- - **checkpoint** (`str`): BLEURT checkpoint. Will default to `BLEURT-tiny` if not specified. Other models that can be chosen are: `"bleurt-tiny-128"`, `"bleurt-tiny-512"`, `"bleurt-base-128"`, `"bleurt-base-512"`, `"bleurt-large-128"`, `"bleurt-large-512"`, `"BLEURT-20-D3"`, `"BLEURT-20-D6"`, `"BLEURT-20-D12"` and `"BLEURT-20"`.
48
 
49
  ### Output Values
50
  - **scores** : a `list` of scores, one per prediction.
 
44
  ### Inputs
45
  - **predictions** (`list` of `str`s): List of generated sentences to score.
46
  - **references** (`list` of `str`s): List of references to compare to.
47
+ - **config_name** (`str`): BLEURT checkpoint. Will default to `BLEURT-tiny` if not specified. Other models that can be chosen are: `"bleurt-tiny-128"`, `"bleurt-tiny-512"`, `"bleurt-base-128"`, `"bleurt-base-512"`, `"bleurt-large-128"`, `"bleurt-large-512"`, `"BLEURT-20-D3"`, `"BLEURT-20-D6"`, `"BLEURT-20-D12"` and `"BLEURT-20"`.
48
 
49
  ### Output Values
50
  - **scores** : a `list` of scores, one per prediction.