fixed nn.py

kmheckel · Feb 17, 2024 · 97ff1f9 · 97ff1f9
1 parent ae2a27d
commit 97ff1f9
Show file tree

Hide file tree

Showing 14 changed files with 3,506 additions and 930 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -14,7 +14,7 @@
 project = 'Spyx'
 copyright = '2023, Kade Heckel'
 author = 'Kade Heckel'
-release = 'v0.1.17'
+release = 'v0.1.18'
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

diff --git a/research/paper/NMNIST_jax.ipynb b/research/paper/NMNIST_jax.ipynb
@@ -210,7 +210,7 @@
     "        grad_params, opt_state = state\n",
     "        events, targets = data # fix this\n",
     "        events = jnp.unpackbits(events, axis=1) # decompress temporal axis\n",
-    "        # compute loss and gradient                    # need better augment rng\n",
+    "        # compute loss and gradient\n",
     "        loss, grads = surrogate_grad(grad_params, events, targets)\n",
     "        # generate updates based on the gradients and optimizer\n",
     "        updates, opt_state = opt.update(grads, opt_state, grad_params)\n",