Fix xlnet run_classifier.

PiperOrigin-RevId: 280516666

Fix xlnet run_classifier.
PiperOrigin-RevId: 280516666
3fab0abd · Hongkun Yu · A. Unique TensorFlower · 3af667fd · 3fab0abd
Commit 3fab0abd authored Nov 14, 2019 by Hongkun Yu Committed by A. Unique TensorFlower Nov 14, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

official/nlp/xlnet/run_classifier.py official/nlp/xlnet/run_classifier.py +1 -3

No files found.
--- a/official/nlp/xlnet/run_classifier.py
+++ b/official/nlp/xlnet/run_classifier.py
@@ -91,9 +91,7 @@ def run_evaluation(strategy,
        _test_step_fn, args=(next(test_iterator),))
    return logits, labels, masks

-  # pylint: disable=protected-access
-  test_iterator = data_utils._get_input_iterator(test_input_fn, strategy)
-  # pylint: enable=protected-access
+  test_iterator = data_utils.get_input_iterator(test_input_fn, strategy)
  correct = 0
  total = 0
  for _ in range(eval_steps):