Skip to content
This repository was archived by the owner on May 10, 2023. It is now read-only.

Commit d6e01c6

Browse files
s1998exelban
authored andcommitted
Fix double softmax in loss issue
1 parent 0311ac5 commit d6e01c6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def model():
5656

5757
fc = tf.layers.dense(inputs=flat, units=1500, activation=tf.nn.relu)
5858
drop = tf.layers.dropout(fc, rate=0.5)
59-
softmax = tf.layers.dense(inputs=drop, units=_NUM_CLASSES, activation=tf.nn.softmax, name=scope.name)
59+
softmax = tf.layers.dense(inputs=drop, units=_NUM_CLASSES, name=scope.name)
6060

6161
y_pred_cls = tf.argmax(softmax, axis=1)
6262

0 commit comments

Comments
 (0)