From 145ca041fbaa7fdda5fd652b1d880bd3d7832e2d Mon Sep 17 00:00:00 2001 From: yick Date: Tue, 5 Nov 2019 16:53:50 +0800 Subject: [PATCH] Update densenet.py fix bug --- DenseNet/densenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/DenseNet/densenet.py b/DenseNet/densenet.py index 88c1937..f0db82b 100644 --- a/DenseNet/densenet.py +++ b/DenseNet/densenet.py @@ -173,7 +173,7 @@ def DenseNet(nb_classes, img_dim, depth, nb_dense_block, growth_rate, dropout_rate=dropout_rate, weight_decay=weight_decay) # add transition - x = transition(x, nb_filter, dropout_rate=dropout_rate, + x = transition(x, concat_axis, nb_filter, dropout_rate=dropout_rate, weight_decay=weight_decay) # The last denseblock does not have a transition