Skip to content

GitLab

"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "f6f96b0521ca87b61199c9ff262ceafa1f3c0f81"
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
import time
# pylint: disable=g-bad-import-order
from absl import flags
from absl.testing import flagsaver
import tensorflow as tf
......@@ -160,9 +161,9 @@ class XLNetSquadAccuracy(XLNetBenchmarkBase):
def __init__(self, output_dir=None, tpu=None, **kwargs):
self.train_data_path = SQUAD_DATA_PATH
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.spiece_model_file = os.path.join(SQUAD_DATA_PATH, "spiece.cased.model")
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.spiece_model_file = os.path.join(SQUAD_DATA_PATH, 'spiece.cased.model')
self.pretrained_checkpoint_path = PRETRAINED_CHECKPOINT_PATH
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