Update eval.py
Browse files
eval.py
CHANGED
@@ -142,7 +142,7 @@ def main(args):
|
|
142 |
dataset = load_dataset(args.dataset, args.config, split=args.split, use_auth_token=True)
|
143 |
|
144 |
# for testing: only process the first two examples as a test
|
145 |
-
dataset = dataset.select(range(10))
|
146 |
|
147 |
# load processor
|
148 |
feature_extractor = AutoFeatureExtractor.from_pretrained(args.model_id)
|
|
|
142 |
dataset = load_dataset(args.dataset, args.config, split=args.split, use_auth_token=True)
|
143 |
|
144 |
# for testing: only process the first two examples as a test
|
145 |
+
# dataset = dataset.select(range(10))
|
146 |
|
147 |
# load processor
|
148 |
feature_extractor = AutoFeatureExtractor.from_pretrained(args.model_id)
|