From f9fd35718a010a0e76326e6fdc1446553d95dd98 Mon Sep 17 00:00:00 2001 From: Hongyi Wang Date: Mon, 23 Apr 2018 15:59:04 -0500 Subject: [PATCH] fix args pass in transition --- 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