Bug fix

AdrianAlan · Jul 20, 2023 · 933f453 · 933f453
1 parent d877ada
commit 933f453
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/part4/4.JetTaggingPointCloud.ipynb b/part4/4.JetTaggingPointCloud.ipynb
@@ -508,14 +508,14 @@
         "        self.attention = tf.keras.layers.MultiHeadAttention(num_heads=self.num_heads, \n",
         "                                                            key_dim=self.hidden_units//self.num_heads)\n",
         "        self.feedforward = tf.keras.Sequential([\n",
-        "            Dense(units=self.mlp_hidden_units, activation=\"relu\"),\n",
+        "            layers.Dense(units=self.mlp_hidden_units, activation=\"relu\"),\n",
         "            # Dropout(rate=self.dropout_rate),\n",
-        "            Dense(units=input_shape[-1])\n",
+        "            layers.Dense(units=input_shape[-1])\n",
         "        ])\n",
-        "        self.layer_norm1 = LayerNormalization(epsilon=1e-6)\n",
-        "        self.layer_norm2 = LayerNormalization(epsilon=1e-6)\n",
-        "        self.dropout1 = Dropout(rate=self.dropout_rate)\n",
-        "        self.dropout2 = Dropout(rate=self.dropout_rate)\n",
+        "        self.layer_norm1 = layers.LayerNormalization(epsilon=1e-6)\n",
+        "        self.layer_norm2 = layers.LayerNormalization(epsilon=1e-6)\n",
+        "        self.dropout1 = layers.Dropout(rate=self.dropout_rate)\n",
+        "        self.dropout2 = layers.Dropout(rate=self.dropout_rate)\n",
         "        super(SABTransformerBlock, self).build(input_shape)\n",
         "        \n",
         "    def call(self, inputs, mask=None):\n",