From 3aeb8db4cd37c179ec1de3f201b8f9c9cfe159f7 Mon Sep 17 00:00:00 2001 From: aleksandrskoselevs Date: Fri, 23 Aug 2024 10:29:52 +0200 Subject: [PATCH] cleaner diff --- Notebooks/Chap09/9_5_Augmentation.ipynb | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/Notebooks/Chap09/9_5_Augmentation.ipynb b/Notebooks/Chap09/9_5_Augmentation.ipynb index 66819ed..bd0c607 100644 --- 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"