diff --git a/agents/ppo/agent.py b/agents/ppo/agent.py index 0f28d7d..82c5f75 100644 --- a/agents/ppo/agent.py +++ b/agents/ppo/agent.py @@ -114,17 +114,6 @@ class Agent: T.nn.utils.clip_grad_norm_( self.critic.parameters(), max_norm=2) - # - # # Calculate the gradient norms for both networks - # actor_grad_norm = T.nn.utils.clip_grad_norm_( - # self.actor.parameters(), max_norm=2) - # - # critic_grad_norm = T.nn.utils.clip_grad_norm_( - # self.critic.parameters(), max_norm=2) - # - # # Log or print the gradient norms - # print(f"Actor Gradient Norm: {actor_grad_norm}") - # print(f"Critic Gradient Norm: {critic_grad_norm}") self.actor.optimizer.step() self.critic.optimizer.step()