Skip to content

Commit

Permalink
Merge pull request WZMIAOMIAO#76 from WZMIAOMIAO/dev
Browse files Browse the repository at this point in the history
add activation for dense
  • Loading branch information
WZMIAOMIAO authored Oct 29, 2020
2 parents 1d9f3a2 + 02c2bf0 commit 96c3475
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensorflow_classification/Test5_resnet/predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()])
Expand Down

0 comments on commit 96c3475

Please sign in to comment.