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

Changed position of comment

Browse Source
This commit is contained in:
Somshubra Majumdar
2017-02-14 12:33:01 -06:00
parent c0bcf2e44f
commit 5c406022c3
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/cifar10_densenet.py
+5 -6
View File
@@ -1,3 +1,8 @@
'''
Trains a DenseNet-40-12 model on the CIFAR-10 Dataset.
Gets a 99.84% accuracy score after 300 epochs.
'''
from __future__ import absolute_import
from __future__ import print_function
from __future__ import division
@@ -13,12 +18,6 @@ from keras.preprocessing.image import ImageDataGenerator
from keras.utils import np_utils
from keras_contrib.applications.densenet import DenseNet
'''
Trains a DenseNet-40-12 model on the CIFAR-10 Dataset.
Gets a 99.84% accuracy score after 300 epochs.
'''
batch_size = 64
nb_classes = 10
nb_epoch = 300