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 e776f61 commit aad3c60Copy full SHA for aad3c60
pytorch_optimizer/optimizer/shampoo.py
@@ -124,6 +124,7 @@ def reset(self):
124
group['betas'][1], # beta2
125
self.inverse_exponent_override,
126
self.block_size,
127
+ self.no_preconditioning_for_layers_with_dim_gt,
128
self.shape_interpretation,
129
self.matrix_eps,
130
self.pre_conditioner_type,
0 commit comments