saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-05 00:00:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update a2c logging

Browse source
This commit is contained in:
Antonin Raffin 2020-03-13 11:48:16 +01:00
parent 29d7018265
commit cf89cac3e9
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/a2c/a2c.py
View file

@ -144,6 +144,8 @@ class A2C(PPO):
explained_var = explained_variance(self.rollout_buffer.returns.flatten(),
self.rollout_buffer.values.flatten())
self._n_updates += 1
logger.logkv("n_updates", self._n_updates)
logger.logkv("explained_variance", explained_var)
logger.logkv("entropy_loss", entropy_loss.item())
logger.logkv("policy_loss", policy_loss.item())