Skip to content

GitLab

Unverified Commit 85d69a7d authored by Hardian Lawi's avatar Hardian Lawi Committed by GitHub
Browse files

Fix missing tpu variable in benchmark_args_tf.py (#13968)

parent 990de2c1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/benchmark/benchmark_args_tf.py
View file @ 85d69a7d
...@@ -79,6 +79,7 @@ class TensorFlowBenchmarkArguments(BenchmarkArguments): ...@@ -79,6 +79,7 @@ class TensorFlowBenchmarkArguments(BenchmarkArguments):
@cached_property @cached_property
@tf_required @tf_required
def _setup_tpu(self) -> Tuple["tf.distribute.cluster_resolver.TPUClusterResolver"]: def _setup_tpu(self) -> Tuple["tf.distribute.cluster_resolver.TPUClusterResolver"]:
tpu = None
if self.tpu: if self.tpu:
try: try:
if self.tpu_name: if self.tpu_name:
......
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