Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "41aafda97ff4b7bc6125c07bbfe3658707414a81"
Commit 5ba3c3f5 authored by David Chen's avatar David Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 265584211
parent f93229b9
Hide 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 ...@@ -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' 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 # pylint: enable=line-too-long
TMP_DIR = os.getenv('TMPDIR')
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
...@@ -116,7 +117,7 @@ class BertSquadBenchmarkReal(BertSquadBenchmarkBase): ...@@ -116,7 +117,7 @@ class BertSquadBenchmarkReal(BertSquadBenchmarkBase):
`benchmark_(number of gpus)_gpu` format. `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) super(BertSquadBenchmarkReal, self).__init__(output_dir=output_dir)
def _setup(self): 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