Skip to content

GitLab

Unverified Commit e92efcf7 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

Mult rouge by 100: standard units (#6359)

parent eda07efa
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/seq2seq/utils.py
View file @ e92efcf7
...@@ -271,7 +271,7 @@ def calculate_rouge(output_lns: List[str], reference_lns: List[str], use_stemmer ...@@ -271,7 +271,7 @@ def calculate_rouge(output_lns: List[str], reference_lns: List[str], use_stemmer
aggregator.add_scores(scores) aggregator.add_scores(scores)
result = aggregator.aggregate() result = aggregator.aggregate()
return {k: v.mid.fmeasure for k, v in result.items()} return {k: v.mid.fmeasure * 100 for k, v in result.items()}
def freeze_params(model: nn.Module): def freeze_params(model: nn.Module):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment