riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-25 23:38:59 +06:00
Code Issues Actions 36 Packages Projects Releases Wiki Activity

fixing again

Browse Source
This commit is contained in:
thomwolf 2019-06-18 16:56:52 +02:00
parent 8bd9118294
commit f55b60b9ee
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
examples/run_classifier.py
View File

@ -420,7 +420,7 @@ def main():
eval_loss = 0 eval_loss = 0
nb_eval_steps = 0 nb_eval_steps = 0
preds = [] preds = []
out_label_ids = [] out_label_ids = None
for input_ids, input_mask, segment_ids, label_ids in tqdm(eval_dataloader, desc="Evaluating"): for input_ids, input_mask, segment_ids, label_ids in tqdm(eval_dataloader, desc="Evaluating"):
input_ids = input_ids.to(device) input_ids = input_ids.to(device)
@ -443,12 +443,12 @@ 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.append(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)
out_label_ids[0] = np.append( out_label_ids = np.append(
out_label_ids[0], label_ids.detach().cpu().numpy(), axis=0) out_label_ids, label_ids.detach().cpu().numpy(), axis=0)
eval_loss = eval_loss / nb_eval_steps eval_loss = eval_loss / nb_eval_steps
preds = preds[0] preds = preds[0]
@ -505,7 +505,7 @@ def main():
eval_loss = 0 eval_loss = 0
nb_eval_steps = 0 nb_eval_steps = 0
preds = [] preds = []
out_label_ids = [] out_label_ids = None
for input_ids, input_mask, segment_ids, label_ids in tqdm(eval_dataloader, desc="Evaluating"): for input_ids, input_mask, segment_ids, label_ids in tqdm(eval_dataloader, desc="Evaluating"):
input_ids = input_ids.to(device) input_ids = input_ids.to(device)
@ -523,13 +523,12 @@ 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.append(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)
out_label_ids[0] = np.append( out_label_ids = np.append(
out_label_ids[0], label_ids.detach().cpu().numpy(), axis=0) out_label_ids, label_ids.detach().cpu().numpy(), axis=0)
eval_loss = eval_loss / nb_eval_steps eval_loss = eval_loss / nb_eval_steps
preds = preds[0] preds = preds[0]