SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

hop

Browse Source
This commit is contained in:
thomwolf
2019-06-18 16:57:15 +02:00
parent f55b60b9ee
commit 4e6edc3274
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/run_classifier.py
View File

@@ -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():
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)