Unverified Commit 79132d4c authored by Jay Zhou's avatar Jay Zhou Committed by GitHub
Browse files

Fix a typo in `examples/pytorch/text-classification/run_classification.py` (#29072)

parent 98308586
...@@ -404,7 +404,7 @@ def main(): ...@@ -404,7 +404,7 @@ def main():
raw_datasets.pop(split) raw_datasets.pop(split)
if data_args.train_split_name is not None: 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["train"] = raw_datasets[data_args.train_split_name]
raw_datasets.pop(data_args.train_split_name) raw_datasets.pop(data_args.train_split_name)
......
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