fix bug

iamhankai · Mar 19, 2019 · a2be08e · a2be08e
1 parent ab32107
commit a2be08e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub_demo.py b/cub_demo.py
@@ -62,7 +62,7 @@ def init_classification(input_fea_map, dim_channel, nb_class, name=None):
     pool = BatchNormalization()(pool)
     pool = Activation('relu')(pool)
     # classification
-    prob = Dropout(dropout)(pool)
+    pool = Dropout(dropout)(pool)
     prob = Dense(nb_class)(pool)
     prob = Activation(activation='softmax',name=name)(prob)
     return prob,pool,fea_map