mirror of
https://github.com/wassname/keras-contrib.git
synced 2026-06-27 16:10:11 +08:00
Correct dense connectivity in DenseNet
This commit is contained in:
@@ -385,7 +385,7 @@ def __dense_block(x, nb_layers, nb_filter, growth_rate, bottleneck=False, dropou
|
||||
conv_block = __conv_block(x, growth_rate, bottleneck, dropout_rate, weight_decay)
|
||||
x_list.append(conv_block)
|
||||
|
||||
x = concatenate([x, conv_block], axis=concat_axis)
|
||||
x = concatenate(x_list, axis=concat_axis)
|
||||
|
||||
if grow_nb_filters:
|
||||
nb_filter += growth_rate
|
||||
|
||||
Reference in New Issue
Block a user