Skip to content

Commit cc392bc

Browse files
committed
Alternative test case
1 parent c69c5a1 commit cc392bc

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pytorch_lightning/plugins/precision/precision_plugin.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,9 @@ def clip_gradients(
121121
if clip_val <= 0:
122122
return
123123

124+
assert gradient_clip_algorithm is GradClipAlgorithmType.VALUE
125+
# gradient_clip_algorithm = GradClipAlgorithmType.VALUE
126+
124127
clip_grad_func = self.clip_grad_funcs[gradient_clip_algorithm]
125128
clip_grad_func(optimizer, clip_val) # type: ignore
126129

0 commit comments

Comments
 (0)