Fixed initialization

lezcano · Jan 30, 2019 · f20526f · f20526f
1 parent f493b78
commit f20526f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
 __pycache__
+mnist
diff --git a/initialization.py b/initialization.py
@@ -11,10 +11,8 @@ def cayley_init(n):
     return create_diag(s, n)
 
 def create_diag(s, n):
-    diag = np.zeros(n)
+    diag = np.zeros(n-1)
     diag[::2] = s
-    if n % 2 == 0:
-        diag = diag[:-1]
     A_init = np.diag(diag, k=1)
     A_init = A_init - A_init.T
     return A_init.astype(np.float32)