Fixing a bug

The function was hard coded to a specific layer. It should generalize to other layers.
hishidar · Mar 20, 2018 · 95fa6e5 · 95fa6e5
1 parent 75a0d88
commit 95fa6e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/building-blocks/NeuronGroups.ipynb b/notebooks/building-blocks/NeuronGroups.ipynb
@@ -441,7 +441,7 @@
         "  # And create a feature visualziation of each group\n",
         "\n",
         "  param_f = lambda: param.image(80, batch=n_groups)\n",
-        "  obj = sum(objectives.direction(\"mixed4d\", channel_factors[i], batch=i)\n",
+        "  obj = sum(objectives.direction(layer, channel_factors[i], batch=i)\n",
         "            for i in range(n_groups))\n",
         "  group_icons = render.render_vis(model, obj, param_f, verbose=False)[-1]\n",
         "  \n",
@@ -1956,4 +1956,4 @@
       "outputs": []
     }
   ]
-}
+}