Skip to content

GitLab

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

Move shakespeare to benchmark folder. 

PiperOrigin-RevId: 304729158
parent 91f6a95b
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/staging/shakespeare/shakespeare_benchmark.py → official/benchmark/shakespeare_benchmark.py
View file @ 17eee4c4
...@@ -23,13 +23,13 @@ import time ...@@ -23,13 +23,13 @@ import time
from absl import flags from absl import flags
import tensorflow as tf # pylint: disable=g-bad-import-order import tensorflow as tf # pylint: disable=g-bad-import-order
from official.staging.shakespeare import shakespeare_main from official.benchmark.models.shakespeare import shakespeare_main
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
from official.utils.misc import keras_utils from official.utils.misc import keras_utils
from official.utils.testing import benchmark_wrappers from official.utils.testing import benchmark_wrappers
from official.utils.testing.perfzero_benchmark import PerfZeroBenchmark from official.utils.testing.perfzero_benchmark import PerfZeroBenchmark
SHAKESPEARE_TRAIN_DATA = 'shakespeare/shakespeare.txt' SHAKESPEARE_TRAIN_DATA = '/placer/prod/home/tensorflow-performance-data/datasets/shakespeare/shakespeare.txt'
TMP_DIR = os.getenv('TMPDIR') TMP_DIR = os.getenv('TMPDIR')
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
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