From 7e69bbab4486e3d0d7b32781a1a5c3e5ea025e14 Mon Sep 17 00:00:00 2001 From: Somshubra Majumdar Date: Wed, 6 Dec 2017 15:02:33 -0600 Subject: [PATCH] Small change to docs --- keras_contrib/applications/nasnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_contrib/applications/nasnet.py b/keras_contrib/applications/nasnet.py index 56f75ed..89ebb98 100644 --- a/keras_contrib/applications/nasnet.py +++ b/keras_contrib/applications/nasnet.py @@ -186,7 +186,7 @@ def NASNet(input_shape=None, img_input = input_tensor assert penultimate_filters % 24 == 0, "`penultimate_filters` needs to be divisible " \ - "by 6 * (2^N)." + "by 24." channel_dim = 1 if K.image_data_format() == 'channels_first' else -1 filters = penultimate_filters // 24