From de4ab68ed2d5f9a63cc6ab18bd911dfb45fb1d5b Mon Sep 17 00:00:00 2001 From: farizrahman4u Date: Sun, 19 Mar 2017 09:14:58 +0530 Subject: [PATCH] update cifar example --- examples/cifar10_densenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/cifar10_densenet.py b/examples/cifar10_densenet.py index db35a89..c495919 100644 --- a/examples/cifar10_densenet.py +++ b/examples/cifar10_densenet.py @@ -26,7 +26,7 @@ img_rows, img_cols = 32, 32 img_channels = 3 # Parameters for the DenseNet model builder -img_dim = (img_channels, img_rows, img_cols) if K.image_dim_ordering() == "th" else (img_rows, img_cols, img_channels) +img_dim = (img_channels, img_rows, img_cols) if K.imgae_data_format() == "channels_first" else (img_rows, img_cols, img_channels) depth = 40 nb_dense_block = 3 growth_rate = 12