From 7438ceb71a2a6c3d9c6772d1145ec7bcaedaa78f Mon Sep 17 00:00:00 2001 From: rubisco0211 <1205456072@qq.com> Date: Sun, 18 Jan 2026 21:10:00 +0800 Subject: [PATCH] fix logical error in gradient shaping --- GradS/fsrl/policy/lagrangian_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GradS/fsrl/policy/lagrangian_base.py b/GradS/fsrl/policy/lagrangian_base.py index 1ab52d2..e45cd93 100644 --- a/GradS/fsrl/policy/lagrangian_base.py +++ b/GradS/fsrl/policy/lagrangian_base.py @@ -250,7 +250,7 @@ def safety_loss(self, values: List, required_GS = True) -> Tuple[torch.tensor, d for index_j in index_list: sim_ij = torch.mean(cos(safety_loss_grad_list[index_i], safety_loss_grad_list[index_j])).item() - sim_list.append(sim_ij) + sim_list.append(sim_ij) if np.max(np.array(sim_list)) < self.sim_threshold and np.min(np.array(sim_list)) > -1 * self.conflicting_threshold: index_list.append(index_i)