Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e2ae236090de86c83f86d094e3862bcdd9548f56"
Commit f41682df authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 305419082
parent eb6fa0b2
Show 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):
default_flags=None,
flag_methods=None,
tpu=None):
# Due to xla legacy benchmark.
tf.compat.v1.enable_v2_behavior()
super(KerasBenchmark, self).__init__(
output_dir=output_dir,
default_flags=default_flags,
......
official/benchmark/transformer_benchmark.py
View file @ f41682df
......@@ -44,8 +44,6 @@ class TransformerBenchmark(PerfZeroBenchmark):
def __init__(self, output_dir=None, default_flags=None, root_data_dir=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 ''
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