Skip to content

Commit

Permalink
No public description
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 666412764
  • Loading branch information
tensorflower-gardener committed Aug 22, 2024
1 parent 43b5203 commit 1667c22
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions official/projects/nhnet/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,9 +183,6 @@ def train(params, strategy, dataset=None):

def run():
"""Runs NHNet using Keras APIs."""
if FLAGS.enable_mlir_bridge:
tf.config.experimental.enable_mlir_bridge()

strategy = distribute_utils.get_distribution_strategy(
distribution_strategy=FLAGS.distribution_strategy, tpu_address=FLAGS.tpu)
if strategy:
Expand Down

0 comments on commit 1667c22

Please sign in to comment.