Default to accuracy metric (#11405)

1ef152eb · Sylvain Gugger · GitHub · e3ff165a · 1ef152eb
Unverified Commit 1ef152eb authored Apr 23, 2021 by Sylvain Gugger Committed by GitHub Apr 23, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

examples/pytorch/text-classification/run_glue_no_trainer.py examples/pytorch/text-classification/run_glue_no_trainer.py +2 -0

No files found.
--- a/examples/pytorch/text-classification/run_glue_no_trainer.py
+++ b/examples/pytorch/text-classification/run_glue_no_trainer.py
@@ -367,6 +367,8 @@ def main():
    # Get the metric function
    if args.task_name is not None:
        metric = load_metric("glue", args.task_name)
+    else:
+        metric = load_metric("accuracy")

    # Train!
    total_batch_size = args.per_device_train_batch_size * accelerator.num_processes * args.gradient_accumulation_steps