Default to accuracy metric (huggingface#11405)

markurtz · Apr 23, 2021 · 1ef152e · 1ef152e
1 parent e3ff165
commit 1ef152e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 preprocess_function(examples):
     # 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