From 92e0ad5aba669a13d37a73dc9efd5faa98ef88ec Mon Sep 17 00:00:00 2001 From: thomwolf Date: Tue, 18 Jun 2019 17:00:52 +0200 Subject: [PATCH] no numpy --- examples/run_classifier.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/run_classifier.py b/examples/run_classifier.py index b2acffe0c3..63cc2b4b9c 100644 --- a/examples/run_classifier.py +++ b/examples/run_classifier.py @@ -456,7 +456,7 @@ def main(): preds = np.argmax(preds, axis=1) elif output_mode == "regression": preds = np.squeeze(preds) - result = compute_metrics(task_name, preds, out_label_ids.numpy()) + result = compute_metrics(task_name, preds, out_label_ids) if args.local_rank != -1: # Average over distributed nodes if needed @@ -533,7 +533,7 @@ def main(): eval_loss = eval_loss / nb_eval_steps preds = preds[0] preds = np.argmax(preds, axis=1) - result = compute_metrics(task_name, preds, out_label_ids.numpy()) + result = compute_metrics(task_name, preds, out_label_ids) if args.local_rank != -1: # Average over distributed nodes if needed