diff --git a/Laboratoire 5.ipynb b/Laboratoire 5.ipynb index 3908fb4..be8ee18 100644 --- a/Laboratoire 5.ipynb +++ b/Laboratoire 5.ipynb @@ -318,8 +318,8 @@ "outputs": [], "source": [ "# TODO Corrigez le code suivant\n", - "conv1 = nn.Conv2d(3, 100, 3, padding=1)\n", - "conv2 = nn.Conv2d(100, 100, 3, padding=1)\n", + "conv1 = nn.Conv2d(3, 32, 3, padding=1)\n", + "conv2 = nn.Conv2d(64, 100, 3, padding=1)\n", "conv2(conv1(image_batch)).shape" ] },