From 024fb66354e903f9f37bc36c0bbe07e16432572c Mon Sep 17 00:00:00 2001
From: Wenda Wu <47494991+wenda-wu@users.noreply.github.com>
Date: Wed, 27 Mar 2019 11:14:25 +0800
Subject: [PATCH] fix missing argument concat_axis

There was a missing argument 'concat_axis' in line 176: x = 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