diff --git a/DenseNet100-cifar10 b/DenseNet100-cifar10 index f262c6b..91beee4 100644 --- a/DenseNet100-cifar10 +++ b/DenseNet100-cifar10 @@ -5,7 +5,7 @@ import os os.environ['TL_BACKEND'] = 'tensorflow' import tensorlayer as tl -from densenet import densenet +from .densenet import densenet tl.logging.set_verbosity(tl.logging.DEBUG) X_train, y_train, X_test, y_test = tl.files.load_cifar10_dataset(shape=(-1, 32, 32, 3), plotable=False)