Merge pull request #3 from Milstein-Lab/gpu

Notebook documentation

Author: yashc73080

EIANN/notebooks/explore_MNIST.ipynb

@@ -62,6 +62,13 @@
     "analyze_receptive_fields = None  # ['H2E']"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Set `train_steps` (above) to 20000 for full training and to replicate results in the paper. Set to 2000 for quick testing."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -74,7 +81,7 @@
     "    projection_config = network_config['projection_config']\n",
     "    training_kwargs = network_config['training_kwargs']\n",
     "\n",
-    "    return Network(layer_config, projection_config, seed=network_seed, **training_kwargs)"
+    "    return Network(layer_config, projection_config, seed=network_seed, device='cpu', **training_kwargs)"
    ]
   },
   {

EIANN/notebooks/explore_spirals.ipynb

@@ -72,7 +72,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "id": "c77d577f",
    "metadata": {},
    "outputs": [
@@ -127,7 +127,7 @@
    ],
    "source": [
     "# Create network object\n",
-    "spiral_net = ut.build_EIANN_from_config(config_file_path, network_seed=network_seed)\n",
+    "spiral_net = ut.build_EIANN_from_config(config_file_path, network_seed=network_seed, device='cpu')\n",
     "\n",
     "# Train network\n",
     "data_generator.manual_seed(data_seed)\n",