diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst index e15bb49..6a769f7 100644 --- a/docs/misc/changelog.rst +++ b/docs/misc/changelog.rst @@ -25,6 +25,7 @@ Others: ^^^^^^^ - Improve typing coverage of the ``VecEnv`` - Removed ``AlreadySteppingError`` and ``NotSteppingError`` that were not used +- Fixed typos in SAC and TD3 Documentation: ^^^^^^^^^^^^^^ diff --git a/stable_baselines3/td3/td3.py b/stable_baselines3/td3/td3.py index 3dc76f7..7738b0b 100644 --- a/stable_baselines3/td3/td3.py +++ b/stable_baselines3/td3/td3.py @@ -152,10 +152,10 @@ class TD3(OffPolicyAlgorithm): target_q = replay_data.rewards + (1 - replay_data.dones) * self.gamma * target_q # Get current Q estimates for each critic network - current_q_esimates = self.critic(replay_data.observations, replay_data.actions) + current_q_estimates = self.critic(replay_data.observations, replay_data.actions) # Compute critic loss - critic_loss = sum([F.mse_loss(current_q, target_q) for current_q in current_q_esimates]) + critic_loss = sum([F.mse_loss(current_q, target_q) for current_q in current_q_estimates]) # Optimize the critics self.critic.optimizer.zero_grad()