PKU-Alignment · Gaiejj · Apr 6, 2023 · Feb 23, 2023 · Feb 27, 2023 · Feb 28, 2023
diff --git a/omnisafe/algorithms/off_policy/sac_lag.py b/omnisafe/algorithms/off_policy/sac_lag.py
@@ -68,8 +68,6 @@ def _log_when_not_update(self) -> None:
         super()._log_when_not_update()
         self._logger.store(
             **{
-                'Loss/Loss_cost_critic': 0.0,
-                'Value/cost_critic': 0.0,
-                'Metrics/LagrangeMultiplier': self._lagrange.lagrangian_multiplier.item(),
+                'Metrics/LagrangeMultiplier': self._lagrange.lagrangian_multiplier.data.item(),
             },
         )
diff --git a/omnisafe/algorithms/off_policy/td3_lag.py b/omnisafe/algorithms/off_policy/td3_lag.py
@@ -61,3 +61,11 @@ def _loss_pi(
         loss_c = self._lagrange.lagrangian_multiplier.item() * loss_q_c
 
         return (loss_r + loss_c).mean() / (1 + self._lagrange.lagrangian_multiplier.item())
+
+    def _log_when_not_update(self) -> None:
+        super()._log_when_not_update()
+        self._logger.store(
+            **{
+                'Metrics/LagrangeMultiplier': self._lagrange.lagrangian_multiplier.data.item(),
+            },
+        )