Update eval.py (#15310)

This commit is contained in:
Patrick von Platen
2022-01-24 11:46:38 +01:00
committed by GitHub
parent b7cb126ccc
commit 4bf97415a4

View File

@@ -68,7 +68,7 @@ def main(args):
dataset = load_dataset(args.dataset, args.config, split=args.split, use_auth_token=True) dataset = load_dataset(args.dataset, args.config, split=args.split, use_auth_token=True)
# for testing: only process the first two examples as a test # for testing: only process the first two examples as a test
dataset = dataset.select(range(10)) # dataset = dataset.select(range(10))
# load processor # load processor
feature_extractor = AutoFeatureExtractor.from_pretrained(args.model_id) feature_extractor = AutoFeatureExtractor.from_pretrained(args.model_id)