Unverified Commit 129fdae0 authored by Theo Linnemann's avatar Theo Linnemann Committed by GitHub
Browse files

Remove reference to args in XLA check (#7344)

Previously, the TFTrainingArguments object did a check to see if XLA was enabled, but did this by referencing `self.args.xla`, when it should be `self.xla`, because it is the args object. This can be verified a few lines above, where the XLA field is set.
parent d2666136
......@@ -119,7 +119,7 @@ class TFTrainingArguments(TrainingArguments):
def _setup_strategy(self) -> Tuple["tf.distribute.Strategy", int]:
logger.info("Tensorflow: setting up strategy")
if self.args.xla:
if self.xla:
tf.config.optimizer.set_jit(True)
gpus = tf.config.list_physical_devices("GPU")
......
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