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

move perfzero benchmark to benchmark/

PiperOrigin-RevId: 305709689
parent 84165751
...@@ -27,7 +27,7 @@ import tensorflow as tf ...@@ -27,7 +27,7 @@ import tensorflow as tf
# pylint: enable=g-bad-import-order # pylint: enable=g-bad-import-order
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
from official.utils.testing.perfzero_benchmark import PerfZeroBenchmark from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
...@@ -19,9 +19,8 @@ from __future__ import division ...@@ -19,9 +19,8 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import tensorflow as tf import tensorflow as tf
from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
from official.utils.testing.perfzero_benchmark import PerfZeroBenchmark
class KerasBenchmark(PerfZeroBenchmark): class KerasBenchmark(PerfZeroBenchmark):
......
...@@ -24,7 +24,7 @@ import tensorflow as tf ...@@ -24,7 +24,7 @@ import tensorflow as tf
from official.vision.image_classification.resnet import common from official.vision.image_classification.resnet import common
from official.vision.image_classification.resnet import resnet_ctl_imagenet_main from official.vision.image_classification.resnet import resnet_ctl_imagenet_main
from official.utils.testing.perfzero_benchmark import PerfZeroBenchmark from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
from official.utils.testing import benchmark_wrappers from official.utils.testing import benchmark_wrappers
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
......
...@@ -27,7 +27,7 @@ from official.benchmark.models.shakespeare import shakespeare_main ...@@ -27,7 +27,7 @@ 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.benchmark.perfzero_benchmark import PerfZeroBenchmark
SHAKESPEARE_TRAIN_DATA = 'shakespeare/shakespeare.txt' SHAKESPEARE_TRAIN_DATA = 'shakespeare/shakespeare.txt'
TMP_DIR = os.getenv('TMPDIR') TMP_DIR = os.getenv('TMPDIR')
......
...@@ -23,7 +23,7 @@ from absl import flags ...@@ -23,7 +23,7 @@ from absl import flags
import tensorflow as tf import tensorflow as tf
import tensorflow_hub as hub import tensorflow_hub as hub
from official.utils.testing.perfzero_benchmark import PerfZeroBenchmark from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
...@@ -22,12 +22,11 @@ import time ...@@ -22,12 +22,11 @@ import time
from absl import flags from absl import flags
import tensorflow as tf import tensorflow as tf
from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
from official.nlp.transformer import misc from official.nlp.transformer import misc
from official.nlp.transformer import transformer_main as transformer_main from official.nlp.transformer import transformer_main as transformer_main
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
from official.utils.testing import benchmark_wrappers from official.utils.testing import benchmark_wrappers
from official.utils.testing.perfzero_benchmark import PerfZeroBenchmark
TRANSFORMER_EN2DE_DATA_DIR_NAME = 'wmt32k-en2de-official' TRANSFORMER_EN2DE_DATA_DIR_NAME = 'wmt32k-en2de-official'
EN2DE_2014_BLEU_DATA_DIR_NAME = 'newstest2014' EN2DE_2014_BLEU_DATA_DIR_NAME = 'newstest2014'
......
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