diff --git a/14_convolutional_neural_networks.ipynb b/14_convolutional_neural_networks.ipynb index a14ec80..bbbfb33 100644 --- a/14_convolutional_neural_networks.ipynb +++ b/14_convolutional_neural_networks.ipynb @@ -588,8 +588,8 @@ "metadata": {}, "outputs": [], "source": [ - "DefaultConv2D = partial(keras.layers.Conv2D,\n", - " kernel_size=3, strides=1, padding=\"SAME\")\n", + "DefaultConv2D = partial(keras.layers.Conv2D, kernel_size=3, strides=1,\n", + " padding=\"SAME\", use_bias=False)\n", "\n", "class ResidualUnit(keras.layers.Layer):\n", " def __init__(self, filters, strides=1, activation=\"relu\", **kwargs):\n", @@ -626,6 +626,8 @@ "model = keras.models.Sequential()\n", "model.add(DefaultConv2D(64, kernel_size=7, strides=2,\n", " input_shape=[224, 224, 3]))\n", + "model.add(keras.layers.BatchNormalization())\n", + "model.add(keras.layers.Activation(\"relu\"))\n", "model.add(keras.layers.MaxPool2D(pool_size=3, strides=2, padding=\"SAME\"))\n", "prev_filters = 64\n", "for filters in [64] * 3 + [128] * 4 + [256] * 6 + [512] * 3:\n",