Removed comments
This commit is contained in:
parent
c278170847
commit
3d7f973789
1 changed files with 0 additions and 11 deletions
|
@ -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()
|
||||
|
|
Loading…
Reference in a new issue