Skip to content

Commit aad3c60

Browse files
committed
fix: missing parameter
1 parent e776f61 commit aad3c60

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pytorch_optimizer/optimizer/shampoo.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ def reset(self):
124124
group['betas'][1], # beta2
125125
self.inverse_exponent_override,
126126
self.block_size,
127+
self.no_preconditioning_for_layers_with_dim_gt,
127128
self.shape_interpretation,
128129
self.matrix_eps,
129130
self.pre_conditioner_type,

0 commit comments

Comments
 (0)