Skip to content

Commit ad368c0

Browse files
committed
Print current algorithm
1 parent cc392bc commit ad368c0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/plugins/precision/precision_plugin.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def clip_gradients(
121121
if clip_val <= 0:
122122
return
123123

124-
assert gradient_clip_algorithm is GradClipAlgorithmType.VALUE
124+
assert gradient_clip_algorithm is GradClipAlgorithmType.VALUE, gradient_clip_algorithm
125125
# gradient_clip_algorithm = GradClipAlgorithmType.VALUE
126126

127127
clip_grad_func = self.clip_grad_funcs[gradient_clip_algorithm]

0 commit comments

Comments
 (0)