Skip to content

GitLab

Commit 3fab0abd authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix xlnet run_classifier. 

PiperOrigin-RevId: 280516666
parent 3af667fd
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
official/nlp/xlnet/run_classifier.py
View file @ 3fab0abd
...@@ -91,9 +91,7 @@ def run_evaluation(strategy, ...@@ -91,9 +91,7 @@ def run_evaluation(strategy,
_test_step_fn, args=(next(test_iterator),)) _test_step_fn, args=(next(test_iterator),))
return logits, labels, masks return logits, labels, masks
# pylint: disable=protected-access test_iterator = data_utils.get_input_iterator(test_input_fn, strategy)
test_iterator = data_utils._get_input_iterator(test_input_fn, strategy)
# pylint: enable=protected-access
correct = 0 correct = 0
total = 0 total = 0
for _ in range(eval_steps): for _ in range(eval_steps):
......
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