From b4dea23e65f8b27968ebf52f83cf854d55b903dc Mon Sep 17 00:00:00 2001 From: Andrew Hundt Date: Fri, 8 Sep 2017 10:28:45 -0400 Subject: [PATCH] cifar10_densenet.py fix parameter bug --- examples/cifar10_densenet.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/examples/cifar10_densenet.py b/examples/cifar10_densenet.py index 79c6c3d..84e2a33 100644 --- a/examples/cifar10_densenet.py +++ b/examples/cifar10_densenet.py @@ -33,8 +33,11 @@ nb_filter = 16 dropout_rate = 0.0 # 0.0 for data augmentation # Create the model (without loading weights) -model = DenseNet(depth, nb_dense_block, growth_rate, nb_filter, dropout_rate=dropout_rate, - input_shape=img_dim, weights=None) +model = DenseNet(depth=depth, nb_dense_block=nb_dense_block, + growth_rate=growth_rate, nb_filter=nb_filter, + dropout_rate=dropout_rate, + input_shape=img_dim, + weights=None) print('Model created') model.summary()