Skip to content

Commit c32d90f

Browse files
authored
fix issue #2
1 parent aa0fcde commit c32d90f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def main(dataset, mode, split):
9898
rgb_holder, is_training=False, dropout_keep_prob=1)
9999
rgb_logits_dropout = tf.nn.dropout(rgb_logits, 1)
100100
rgb_fc_out = tf.layers.dense(
101-
rgb_logits_dropout, _CLASS_NUM[dataset], tf.nn.relu, use_bias=True)
101+
rgb_logits_dropout, _CLASS_NUM[dataset], use_bias=True)
102102
rgb_top_1_op = tf.nn.in_top_k(rgb_fc_out, label_holder, 1)
103103
if mode in ['flow', 'mixed']:
104104
with tf.variable_scope(_SCOPE['flow']):

0 commit comments

Comments
 (0)