From f55b60b9ee9bc8f7f8ecf04f5d53d0417fbce3d8 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Tue, 18 Jun 2019 16:56:52 +0200 Subject: [PATCH] fixing again --- examples/run_classifier.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/examples/run_classifier.py b/examples/run_classifier.py index 90f9e08ebc9..2667b82d727 100644 --- a/examples/run_classifier.py +++ b/examples/run_classifier.py @@ -420,7 +420,7 @@ def main(): eval_loss = 0 nb_eval_steps = 0 preds = [] - out_label_ids = [] + out_label_ids = None for input_ids, input_mask, segment_ids, label_ids in tqdm(eval_dataloader, desc="Evaluating"): input_ids = input_ids.to(device) @@ -443,12 +443,12 @@ def main(): nb_eval_steps += 1 if len(preds) == 0: preds.append(logits.detach().cpu().numpy()) - out_label_ids.append(label_ids.detach().cpu().numpy()) + out_label_ids = label_ids.detach().cpu().numpy()) else: preds[0] = np.append( preds[0], logits.detach().cpu().numpy(), axis=0) - out_label_ids[0] = np.append( - out_label_ids[0], label_ids.detach().cpu().numpy(), axis=0) + out_label_ids = np.append( + out_label_ids, label_ids.detach().cpu().numpy(), axis=0) eval_loss = eval_loss / nb_eval_steps preds = preds[0] @@ -505,7 +505,7 @@ def main(): eval_loss = 0 nb_eval_steps = 0 preds = [] - out_label_ids = [] + out_label_ids = None for input_ids, input_mask, segment_ids, label_ids in tqdm(eval_dataloader, desc="Evaluating"): input_ids = input_ids.to(device) @@ -523,13 +523,12 @@ def main(): nb_eval_steps += 1 if len(preds) == 0: preds.append(logits.detach().cpu().numpy()) - out_label_ids.append(label_ids.detach().cpu().numpy()) + out_label_ids = label_ids.detach().cpu().numpy()) else: preds[0] = np.append( preds[0], logits.detach().cpu().numpy(), axis=0) - out_label_ids[0] = np.append( - out_label_ids[0], label_ids.detach().cpu().numpy(), axis=0) - + out_label_ids = np.append( + out_label_ids, label_ids.detach().cpu().numpy(), axis=0) eval_loss = eval_loss / nb_eval_steps preds = preds[0]