From a209520b0c925311c4eb5f40d4e64bdc9a3ecc25 Mon Sep 17 00:00:00 2001 From: Antonin RAFFIN Date: Sat, 26 Sep 2020 18:26:54 +0200 Subject: [PATCH] Separate checks --- stable_baselines3/tqc/tqc.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/stable_baselines3/tqc/tqc.py b/stable_baselines3/tqc/tqc.py index 312ce0c..7c386f1 100644 --- a/stable_baselines3/tqc/tqc.py +++ b/stable_baselines3/tqc/tqc.py @@ -277,9 +277,10 @@ class TQC(OffPolicyAlgorithm): logger.record("train/critic_loss", np.mean(critic_losses)) if len(ent_coef_losses) > 0: logger.record("train/ent_coef_loss", np.mean(ent_coef_losses)) - if hasattr(self.actor, "scale_expert"): - with th.no_grad(): + with th.no_grad(): + if hasattr(self.actor, "scale_expert"): logger.record("train/scale_expert", th.sigmoid(self.actor.scale_expert).mean().item()) + if hasattr(self.actor, "scale_rl"): logger.record("train/scale_rl", th.sigmoid(self.actor.scale_rl).mean().item()) def pretrain(