Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 305419082
parent eb6fa0b2
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
official/benchmark/keras_benchmark.py
View file @ f41682df
...@@ -32,8 +32,6 @@ class KerasBenchmark(PerfZeroBenchmark): ...@@ -32,8 +32,6 @@ class KerasBenchmark(PerfZeroBenchmark):
default_flags=None, default_flags=None,
flag_methods=None, flag_methods=None,
tpu=None): tpu=None):
# Due to xla legacy benchmark.
tf.compat.v1.enable_v2_behavior()
super(KerasBenchmark, self).__init__( super(KerasBenchmark, self).__init__(
output_dir=output_dir, output_dir=output_dir,
default_flags=default_flags, default_flags=default_flags,
......
official/benchmark/transformer_benchmark.py
View file @ f41682df
...@@ -44,8 +44,6 @@ class TransformerBenchmark(PerfZeroBenchmark): ...@@ -44,8 +44,6 @@ class TransformerBenchmark(PerfZeroBenchmark):
def __init__(self, output_dir=None, default_flags=None, root_data_dir=None, def __init__(self, output_dir=None, default_flags=None, root_data_dir=None,
flag_methods=None): flag_methods=None):
# Due to xla legacy benchmark.
tf.compat.v1.enable_v2_behavior()
root_data_dir = root_data_dir if root_data_dir else '' root_data_dir = root_data_dir if root_data_dir else ''
self.train_data_dir = os.path.join(root_data_dir, self.train_data_dir = os.path.join(root_data_dir,
......
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