Skip to content

GitLab

Commit 5ba3c3f5 authored by David Chen's avatar David Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 265584211
parent f93229b9
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/bert/benchmark/bert_squad_benchmark.py
View file @ 5ba3c3f5
......@@ -43,6 +43,7 @@ SQUAD_FULL_INPUT_META_DATA_PATH = 'gs://tf-perfzero-data/bert/squad/squad_full_m
MODEL_CONFIG_FILE_PATH = 'gs://cloud-tpu-checkpoints/bert/tf_20/uncased_L-24_H-1024_A-16/bert_config'
# pylint: enable=line-too-long
TMP_DIR = os.getenv('TMPDIR')
FLAGS = flags.FLAGS
......@@ -116,7 +117,7 @@ class BertSquadBenchmarkReal(BertSquadBenchmarkBase):
`benchmark_(number of gpus)_gpu` format.
"""
def __init__(self, output_dir=None, **kwargs):
def __init__(self, output_dir=TMP_DIR, **kwargs):
super(BertSquadBenchmarkReal, self).__init__(output_dir=output_dir)
def _setup(self):
......
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