(self)
Trains RL for one epoch.
Trains RL for one epoch.
Trains RL for one epoch .
def train_epoch(self):
    """Trains RL for one epoch."""
    # When restoring, calculate how many evals are remaining.
    n_evals = remaining_evals(
        self._policy_trainer.step,
        self._epoch,
        self._policy_train_steps_per_epoch,
        self._policy_evals_per_epoch)
    for _ in range(n_evals):
      self._policy_trainer.train_epoch(
          self._policy_train_steps_per_epoch // self._policy_evals_per_epoch,
          self._policy_eval_steps)
def train_epoch ( self ) : # When restoring, calculate how many evals are remaining. n_evals = remaining_evals ( self . _policy_trainer . step , self . _epoch , self . _policy_train_steps_per_epoch , self . _policy_evals_per_epoch ) for _ in range ( n_evals ) : self . _policy_trainer . train_epoch ( self . _policy_train_steps_per_epoch // self . _policy_evals_per_epoch , self . _policy_eval_steps )