wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-06-26 16:00:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

cifar10_densenet.py fix parameter bug

Browse Source
This commit is contained in:
Andrew Hundt
2017-09-08 10:28:45 -04:00
parent b185d2655b
commit b4dea23e65
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/cifar10_densenet.py
+5 -2
View File
@@ -33,8 +33,11 @@ nb_filter = 16
dropout_rate = 0.0 # 0.0 for data augmentation
# Create the model (without loading weights)
model = DenseNet(depth, nb_dense_block, growth_rate, nb_filter, dropout_rate=dropout_rate,
input_shape=img_dim, weights=None)
model = DenseNet(depth=depth, nb_dense_block=nb_dense_block,
growth_rate=growth_rate, nb_filter=nb_filter,
dropout_rate=dropout_rate,
input_shape=img_dim,
weights=None)
print('Model created')
model.summary()