cleaner diff

sid0sid · Aug 23, 2024 · 3aeb8db · 3aeb8db
1 parent 305a055
commit 3aeb8db
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/Notebooks/Chap09/9_5_Augmentation.ipynb b/Notebooks/Chap09/9_5_Augmentation.ipynb
@@ -107,10 +107,7 @@
         "  # Initialize the parameters with He initialization\n",
         "  if isinstance(layer_in, nn.Linear):\n",
         "    nn.init.kaiming_uniform_(layer_in.weight)\n",
-        "    layer_in.bias.data.fill_(0.0)\n",
-        "\n",
-        "# Call the function you just defined\n",
-        "model.apply(weights_init)"
+        "    layer_in.bias.data.fill_(0.0)\n"
       ],
       "metadata": {
         "id": "JfIFWFIL33eF"