add activation for dense

linhuangwei · Oct 29, 2020 · 02c2bf0 · 02c2bf0
1 parent 72af8c2
commit 02c2bf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_classification/Test5_resnet/predict.py b/tensorflow_classification/Test5_resnet/predict.py
@@ -37,7 +37,7 @@
 model = tf.keras.Sequential([feature,
                              tf.keras.layers.GlobalAvgPool2D(),
                              tf.keras.layers.Dropout(rate=0.5),
-                             tf.keras.layers.Dense(1024),
+                             tf.keras.layers.Dense(1024, activation="relu"),
                              tf.keras.layers.Dropout(rate=0.5),
                              tf.keras.layers.Dense(5),
                              tf.keras.layers.Softmax()])