We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cc392bc commit ad368c0Copy full SHA for ad368c0
pytorch_lightning/plugins/precision/precision_plugin.py
@@ -121,7 +121,7 @@ def clip_gradients(
121
if clip_val <= 0:
122
return
123
124
- assert gradient_clip_algorithm is GradClipAlgorithmType.VALUE
+ assert gradient_clip_algorithm is GradClipAlgorithmType.VALUE, gradient_clip_algorithm
125
# gradient_clip_algorithm = GradClipAlgorithmType.VALUE
126
127
clip_grad_func = self.clip_grad_funcs[gradient_clip_algorithm]
0 commit comments