Removed comments

This commit is contained in:
Vasilis Valatsos 2023-12-10 20:16:00 +01:00
parent c278170847
commit 3d7f973789

View file

@ -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()