b1d9ac5b8b4d9e47ae146c77347d99e28e9a2701,official/transformer/v2/transformer_main.py,TransformerTask,train,#TransformerTask#,153

Before Change

  def train(self):
    Trains the model.
    params, flags_obj, is_train = self.params, self.flags_obj, True
    // Sets config options.
    keras_utils.set_session_config(
        enable_xla=flags_obj.enable_xla)

After Change

        // Runs training steps.
        train_steps(iter(train_ds), train_steps_per_eval)
        train_loss = train_loss_metric.result().numpy().astype(float)
        logging.info("Train Step: %d/%d / loss = %s",
                     i * flags_obj.steps_between_evals, flags_obj.train_steps,
                     train_loss)

Instances