SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update run_glue_no_trainer.py (#17546)

Browse Source
This commit is contained in:
bhuang
2022-06-03 18:29:37 +02:00
committed by GitHub
parent 8343901263
commit 254d9c068e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/text-classification/run_glue_no_trainer.py
View File

@@ -270,7 +270,7 @@ def main():
data_files["train"] = args.train_file data_files["train"] = args.train_file
if args.validation_file is not None: if args.validation_file is not None:
data_files["validation"] = args.validation_file data_files["validation"] = args.validation_file
extension = (args.train_file if args.train_file is not None else args.valid_file).split(".")[-1] extension = (args.train_file if args.train_file is not None else args.validation_file).split(".")[-1]
raw_datasets = load_dataset(extension, data_files=data_files) raw_datasets = load_dataset(extension, data_files=data_files)
# See more about loading any type of standard or custom dataset at # See more about loading any type of standard or custom dataset at
# https://huggingface.co/docs/datasets/loading_datasets.html. # https://huggingface.co/docs/datasets/loading_datasets.html.