remove unused return value

iProzd · Nov 27, 2024 · f7265a3 · f7265a3
1 parent b73e5b1
commit f7265a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
@@ -689,7 +689,7 @@ def step(_step_id, task_key="Default") -> None:
                 )
                 loss.backward()
                 if self.gradient_max_norm > 0.0:
-                    grad_norm = torch.nn.utils.clip_grad_norm_(
+                    torch.nn.utils.clip_grad_norm_(
                         self.wrapper.parameters(),
                         self.gradient_max_norm,
                         error_if_nonfinite=True,