[DEL] Remove asserts

f-dangel · Nov 7, 2023 · ca547e5 · ca547e5
1 parent 3597fde
commit ca547e5
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/curvlinops/kfac.py b/curvlinops/kfac.py
@@ -204,8 +204,6 @@ def _matvec(self, x: ndarray) -> ndarray:
 
                 w_cols = mod.weight.shape[1]
                 x_torch[w_pos], x_torch[b_pos] = x_joint.split([w_cols, 1], dim=1)
-                assert x_torch.w_pos.shape == mod.weight.shape
-                assert x_torch.b_pos.shape == mod.bias.shape
 
             # for weights we need to multiply from the right with aaT
             # for weights and biases we need to multiply from the left with ggT