We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4728256 commit b16a127Copy full SHA for b16a127
examples/models/core/llama/summarize_long.py
@@ -397,12 +397,11 @@ def main(args):
397
beam_idx].compute()
398
for key in computed_metrics_tensorrt_llm.keys():
399
logger.info(
400
- f' {key} : {computed_metrics_tensorrt_llm[key].mid[2]*100}'
401
- )
+ f' {key} : {computed_metrics_tensorrt_llm[key]*100}')
402
403
if args.check_accuracy and beam_idx == 0:
404
- assert computed_metrics_tensorrt_llm['rouge1'].mid[
405
- 2] * 100 > args.tensorrt_llm_rouge1_threshold
+ assert computed_metrics_tensorrt_llm[
+ 'rouge1'] * 100 > args.tensorrt_llm_rouge1_threshold
406
407
408
if __name__ == '__main__':
0 commit comments