From 96887b2ee6bdad1577993c470144db607f11eaba Mon Sep 17 00:00:00 2001 From: Somshubra Majumdar Date: Mon, 13 Feb 2017 00:14:17 -0600 Subject: [PATCH] Corrected a few mistakes --- keras_contrib/applications/densenet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/keras_contrib/applications/densenet.py b/keras_contrib/applications/densenet.py index 6ad0a1d..539dd0b 100644 --- a/keras_contrib/applications/densenet.py +++ b/keras_contrib/applications/densenet.py @@ -40,7 +40,7 @@ def DenseNet(depth=40, nb_dense_block=3, growth_rate=12, nb_filter=16, classes=10): """Instantiate the DenseNet architecture, optionally loading weights pre-trained - on CIFAR-100. Note that when using TensorFlow, + on CIFAR-10. Note that when using TensorFlow, for best performance you should set `image_dim_ordering="tf"` in your Keras config at ~/.keras/keras.json. @@ -269,6 +269,7 @@ def __create_dense_net(nb_classes, img_input, include_top, depth=40, nb_dense_bl Args: nb_classes: number of classes img_input: tuple of shape (channels, rows, columns) or (rows, columns, channels) + include_top: flag to include the final Dense layer depth: number or layers nb_dense_block: number of dense blocks to add to end (generally = 3) growth_rate: number of filters to add per dense block