Pneuma/pneuma-pygame
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Removed comments

Browse source
This commit is contained in:
Vasilis Valatsos 2023-12-10 20:16:00 +01:00
parent c278170847
commit 3d7f973789
1 changed files with 0 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
agents/ppo/agent.py
View file

@ -114,17 +114,6 @@ class Agent:
T.nn.utils.clip_grad_norm_( T.nn.utils.clip_grad_norm_(
self.critic.parameters(), max_norm=2) 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.actor.optimizer.step()
self.critic.optimizer.step() self.critic.optimizer.step()