Skip to content

Commit b16a127

Browse files
authored
fixing the metric fmeasure access (#3774)
Signed-off-by: Rakib Hasan <rhasan@nvidia.com>
1 parent 4728256 commit b16a127

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

examples/models/core/llama/summarize_long.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -397,12 +397,11 @@ def main(args):
397397
beam_idx].compute()
398398
for key in computed_metrics_tensorrt_llm.keys():
399399
logger.info(
400-
f' {key} : {computed_metrics_tensorrt_llm[key].mid[2]*100}'
401-
)
400+
f' {key} : {computed_metrics_tensorrt_llm[key]*100}')
402401

403402
if args.check_accuracy and beam_idx == 0:
404-
assert computed_metrics_tensorrt_llm['rouge1'].mid[
405-
2] * 100 > args.tensorrt_llm_rouge1_threshold
403+
assert computed_metrics_tensorrt_llm[
404+
'rouge1'] * 100 > args.tensorrt_llm_rouge1_threshold
406405

407406

408407
if __name__ == '__main__':

0 commit comments

Comments
 (0)