fix typo

2018-11-03 17:10:23 -04:00
parent 5858e8e4dd
commit 1ba5b58c20
1 changed files with 1 additions and 1 deletions
--- a/run_classifier_pytorch.py
+++ b/run_classifier_pytorch.py
@@ -571,7 +571,7 @@ def main():

        model.eval()
        eval_loss, eval_accuracy = 0, 0
-        nb_eval_steps, nb_eval_examples = 0
+        nb_eval_steps, nb_eval_examples = 0, 0
        for input_ids, input_mask, segment_ids, label_ids in eval_dataloader:
            input_ids = input_ids.to(device)
            input_mask = input_mask.float().to(device)