Remove explicit initialize_tpu_system call from model garden.

PiperOrigin-RevId: 290795650

Remove explicit initialize_tpu_system call from model garden.
PiperOrigin-RevId: 290795650
ac97f01b · Peter Hawkins · A. Unique TensorFlower · d206786b · ac97f01b
Commit ac97f01b authored Jan 21, 2020 by Peter Hawkins Committed by A. Unique TensorFlower Jan 21, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

official/utils/misc/tpu_lib.py official/utils/misc/tpu_lib.py +1 -0

No files found.
--- a/official/utils/misc/tpu_lib.py
+++ b/official/utils/misc/tpu_lib.py
@@ -30,4 +30,5 @@ def tpu_initialize(tpu_address):
      tpu=tpu_address)
  if tpu_address not in ('', 'local'):
    tf.config.experimental_connect_to_cluster(cluster_resolver)
+  tf.tpu.experimental.initialize_tpu_system(cluster_resolver)
  return cluster_resolver