diff --git a/examples/cifar10_densenet.py b/examples/cifar10_densenet.py index 669105a..ffc3930 100644 --- a/examples/cifar10_densenet.py +++ b/examples/cifar10_densenet.py @@ -16,7 +16,7 @@ from keras.datasets import cifar10 from keras.optimizers import Adam from keras.preprocessing.image import ImageDataGenerator from keras.utils import np_utils -from keras_contrib.applications.densenet import DenseNet +from keras_contrib.applications import DenseNet batch_size = 64 nb_classes = 10 diff --git a/examples/cifar10_ror.py b/examples/cifar10_ror.py index ab1e0a6..ca9461f 100644 --- a/examples/cifar10_ror.py +++ b/examples/cifar10_ror.py @@ -12,7 +12,7 @@ from keras.datasets import cifar10 from keras.preprocessing.image import ImageDataGenerator from keras.optimizers import Adam -from keras_contrib.applications.ror import ResidualOfResidual +from keras_contrib.applications import ResidualOfResidual batch_size = 64 nb_epoch = 150 diff --git a/keras_contrib/applications/__init__.py b/keras_contrib/applications/__init__.py index e69de29..e9d829d 100644 --- a/keras_contrib/applications/__init__.py +++ b/keras_contrib/applications/__init__.py @@ -0,0 +1,2 @@ +from .densenet import DenseNet +from .ror import ResidualOfResidual