Update module.py

amorehead · web-flow · commit de84676cdede · 2025-09-10T10:14:45.000-07:00
diff --git a/src/lightning/pytorch/core/module.py b/src/lightning/pytorch/core/module.py
@@ -1274,7 +1274,7 @@ def clip_gradients(
 
         gradient_clip_algorithm = GradClipAlgorithmType(gradient_clip_algorithm)
         self.trainer.precision_plugin.clip_gradients(
-            self.trainer.model, optimizer, gradient_clip_val, gradient_clip_algorithm
+            optimizer, gradient_clip_val, gradient_clip_algorithm, module=self.trainer.model,
         )
 
     def configure_gradient_clipping(

Original file line number	Diff line number	Diff line change
`@@ -1274,7 +1274,7 @@ def clip_gradients(`
`1274`	`1274`
`1275`	`1275`	`gradient_clip_algorithm = GradClipAlgorithmType(gradient_clip_algorithm)`
`1276`	`1276`	`self.trainer.precision_plugin.clip_gradients(`
`1277`		`- self.trainer.model, optimizer, gradient_clip_val, gradient_clip_algorithm`
	`1277`	`+ optimizer, gradient_clip_val, gradient_clip_algorithm, module=self.trainer.model,`
`1278`	`1278`	`)`
`1279`	`1279`
`1280`	`1280`	`def configure_gradient_clipping(`