Skip to content

GitLab

Commit 4e6edc32 authored by thomwolf's avatar thomwolf
Browse files

hop

parent f55b60b9
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/run_classifier.py
View file @ 4e6edc32
...@@ -443,7 +443,7 @@ def main(): ...@@ -443,7 +443,7 @@ def main():
nb_eval_steps += 1 nb_eval_steps += 1
if len(preds) == 0: if len(preds) == 0:
preds.append(logits.detach().cpu().numpy()) preds.append(logits.detach().cpu().numpy())
out_label_ids = label_ids.detach().cpu().numpy()) out_label_ids = label_ids.detach().cpu().numpy()
else: else:
preds[0] = np.append( preds[0] = np.append(
preds[0], logits.detach().cpu().numpy(), axis=0) preds[0], logits.detach().cpu().numpy(), axis=0)
...@@ -523,7 +523,7 @@ def main(): ...@@ -523,7 +523,7 @@ def main():
nb_eval_steps += 1 nb_eval_steps += 1
if len(preds) == 0: if len(preds) == 0:
preds.append(logits.detach().cpu().numpy()) preds.append(logits.detach().cpu().numpy())
out_label_ids = label_ids.detach().cpu().numpy()) out_label_ids = label_ids.detach().cpu().numpy()
else: else:
preds[0] = np.append( preds[0] = np.append(
preds[0], logits.detach().cpu().numpy(), axis=0) preds[0], logits.detach().cpu().numpy(), axis=0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment