2c0f1b2d13e090f3ca1b23254cfe0391f3f3894e,tf_agents/agents/behavioral_cloning/behavioral_cloning_agent.py,BehavioralCloningAgent,_train,#BehavioralCloningAgent#Any#Any#,214

Before Change

        variables_to_train=lambda: self._cloning_network.trainable_weights,
    )

    return loss_info

  @eager_utils.future_in_eager_mode
  // TODO(b/79688437): Figure out how to enable defun for Eager mode.
  // @tfe.defun

After Change

    self._apply_loss(aggregated_losses, self._cloning_network.trainable_weights,
                     tape, self._optimizer)
    self.train_step_counter.assign_add(1)
    return tf_agent.LossInfo(aggregated_losses.total_loss,
                             BehavioralCloningLossInfo(per_example_loss))

Instances