Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 326286926
parent 4ad4a3cc
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
official/benchmark/xlnet_benchmark.py
View file @ 94561082
...@@ -23,6 +23,7 @@ import os ...@@ -23,6 +23,7 @@ import os
import time import time
# pylint: disable=g-bad-import-order # pylint: disable=g-bad-import-order
from absl import flags from absl import flags
from absl.testing import flagsaver from absl.testing import flagsaver
import tensorflow as tf import tensorflow as tf
...@@ -160,9 +161,9 @@ class XLNetSquadAccuracy(XLNetBenchmarkBase): ...@@ -160,9 +161,9 @@ class XLNetSquadAccuracy(XLNetBenchmarkBase):
def __init__(self, output_dir=None, tpu=None, **kwargs): def __init__(self, output_dir=None, tpu=None, **kwargs):
self.train_data_path = SQUAD_DATA_PATH self.train_data_path = SQUAD_DATA_PATH
self.predict_file = os.path.join(SQUAD_DATA_PATH, "dev-v2.0.json") self.predict_file = os.path.join(SQUAD_DATA_PATH, 'dev-v2.0.json')
self.test_data_path = os.path.join(SQUAD_DATA_PATH, "12048.eval.tf_record") self.test_data_path = os.path.join(SQUAD_DATA_PATH, '12048.eval.tf_record')
self.spiece_model_file = os.path.join(SQUAD_DATA_PATH, "spiece.cased.model") self.spiece_model_file = os.path.join(SQUAD_DATA_PATH, 'spiece.cased.model')
self.pretrained_checkpoint_path = PRETRAINED_CHECKPOINT_PATH self.pretrained_checkpoint_path = PRETRAINED_CHECKPOINT_PATH
super(XLNetSquadAccuracy, self).__init__(output_dir=output_dir, tpu=tpu) super(XLNetSquadAccuracy, self).__init__(output_dir=output_dir, tpu=tpu)
......
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