update week 15

Author: mhjensen

doc/pub/week15/ipynb/week15.ipynb

@@ -91,10 +91,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "a15172b8",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "PyTorch 2.3.0.post100 | device: cpu\n",
+      "Train: 60000 | Test: 10000\n",
+      "Image shape: torch.Size([1, 28, 28])\n",
+      "Loaders ready (num_workers=0)\n"
+     ]
+    }
+   ],
    "source": [
     "# ── Standard imports ──────────────────────────────────────────────────────────\n",
     "import math, time, os\n",
@@ -166,10 +177,36 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "f3df2b1e",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "VAE parameters: 258,025\n",
+      "VAE(\n",
+      "  (encoder): Sequential(\n",
+      "    (0): Conv2d(1, 32, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))\n",
+      "    (1): LeakyReLU(negative_slope=0.2)\n",
+      "    (2): Conv2d(32, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))\n",
+      "    (3): LeakyReLU(negative_slope=0.2)\n",
+      "    (4): Flatten(start_dim=1, end_dim=-1)\n",
+      "  )\n",
+      "  (fc_mu): Linear(in_features=3136, out_features=20, bias=True)\n",
+      "  (fc_log_var): Linear(in_features=3136, out_features=20, bias=True)\n",
+      "  (fc_dec): Linear(in_features=20, out_features=3136, bias=True)\n",
+      "  (decoder): Sequential(\n",
+      "    (0): Unflatten(dim=1, unflattened_size=(64, 7, 7))\n",
+      "    (1): ConvTranspose2d(64, 32, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))\n",
+      "    (2): LeakyReLU(negative_slope=0.2)\n",
+      "    (3): ConvTranspose2d(32, 1, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))\n",
+      "  )\n",
+      ")\n"
+     ]
+    }
+   ],
    "source": [
     "# ── VAE Model ─────────────────────────────────────────────────────────────\n",
     "class VAE(nn.Module):\n",
@@ -277,7 +314,16 @@
    "execution_count": null,
    "id": "409ab2e8",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Training VAE on MNIST (20 epochs)...\n",
+      "(Uses [0,1]-normalised pixels and Bernoulli decoder)\n"
+     ]
+    }
+   ],
    "source": [
     "def train_vae(model, loader, epochs=20, lr=1e-3, beta=1.0):\n",
     "    \"\"\"\n",

doc/pub/week15/pdf/addition2week15.pdf

@@ -0,0 +1 @@
+{'holder': ['Morten Hjorth-Jensen'], 'year': '1999-2026', 'license': 'Released under CC Attribution-NonCommercial 4.0 license', 'cite doconce': False}