From 79132d4cfe42eca5812e8c45ea1b075f04f907b6 Mon Sep 17 00:00:00 2001 From: Jay Zhou <50169346+Ja1Zhou@users.noreply.github.com> Date: Mon, 19 Feb 2024 05:01:15 -0800 Subject: [PATCH] Fix a typo in `examples/pytorch/text-classification/run_classification.py` (#29072) --- examples/pytorch/text-classification/run_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/text-classification/run_classification.py b/examples/pytorch/text-classification/run_classification.py index c0e4c11374..ceb16f14ec 100755 --- a/examples/pytorch/text-classification/run_classification.py +++ b/examples/pytorch/text-classification/run_classification.py @@ -404,7 +404,7 @@ def main(): raw_datasets.pop(split) if data_args.train_split_name is not None: - logger.info(f"using {data_args.validation_split_name} as validation set") + logger.info(f"using {data_args.train_split_name} as train set") raw_datasets["train"] = raw_datasets[data_args.train_split_name] raw_datasets.pop(data_args.train_split_name)