mirror of
https://github.com/wassname/keras-contrib.git
synced 2026-06-27 16:10:11 +08:00
Small change to docs
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user