def METHOD_NAME(self, total_steps: int, update_steps: int, train_epochs: int = 1) -> None:
    self._on_fit_start()
    for i in tqdm(range(total_steps // update_steps), desc='Trainer', disable=not is_rank_0()):
        self._on_episode_start(i)
        self._learn(update_steps, train_epochs)
        self._on_update_start()
        self._update_remote_makers()
        self._on_update_end()
        self._on_episode_end(i)
    self._on_fit_end()