Skip to content

Commit

Permalink
fix: dropout layers in lab-10-7-mnist_nn_higher_level_API.py (hunkim#170
Browse files Browse the repository at this point in the history
)

- Layers were not connected correctly
  • Loading branch information
Ggoals authored and kkweon committed May 25, 2017
1 parent 566d851 commit 3825615
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions lab-10-7-mnist_nn_higher_level_API.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,11 +47,11 @@
hidden_layer1 = fully_connected(X, hidden_output_size, scope="h1")
h1_drop = dropout(hidden_layer1, keep_prob, is_training=train_mode)
hidden_layer2 = fully_connected(h1_drop, hidden_output_size, scope="h2")
h2_drop = dropout(hidden_layer1, keep_prob, is_training=train_mode)
h2_drop = dropout(hidden_layer2, keep_prob, is_training=train_mode)
hidden_layer3 = fully_connected(h2_drop, hidden_output_size, scope="h3")
h3_drop = dropout(hidden_layer1, keep_prob, is_training=train_mode)
h3_drop = dropout(hidden_layer3, keep_prob, is_training=train_mode)
hidden_layer4 = fully_connected(h3_drop, hidden_output_size, scope="h4")
h4_drop = dropout(hidden_layer1, keep_prob, is_training=train_mode)
h4_drop = dropout(hidden_layer4, keep_prob, is_training=train_mode)
hypothesis = fully_connected(h4_drop, final_output_size, activation_fn=None, scope="hypothesis")


Expand Down

0 comments on commit 3825615

Please sign in to comment.