From cea8e22977349a3400f53b783a0fe34c4985a5ba Mon Sep 17 00:00:00 2001 From: Xiao Wang Date: Thu, 30 Mar 2023 11:38:00 +0800 Subject: [PATCH] remove useless statement --- src/run_uie.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/run_uie.py b/src/run_uie.py index 7cc6627..207e5c8 100644 --- a/src/run_uie.py +++ b/src/run_uie.py @@ -497,7 +497,6 @@ def compute_rouge_metrics(dataset, preds, save_prefix=None): num_beams = data_args.num_beams if data_args.num_beams is not None else training_args.generation_num_beams repetition_penalty = data_args.repetition_penalty - # TODO, test debug, bloomz, flan-t5 if training_args.do_predict: logger.info("*** Prediction ***") logger.info("*** Loading CheckPoint ***")