diff --git a/DenseNet121-ImageNet b/DenseNet121-ImageNet index 41c2293..c550008 100644 --- a/DenseNet121-ImageNet +++ b/DenseNet121-ImageNet @@ -5,7 +5,7 @@ import os os.environ['TL_BACKEND'] = 'tensorflow' import tensorlayer as tl -from DenseNet_tensorlayer import densenet +from densenet import densenet tl.logging.set_verbosity(tl.logging.DEBUG) def load_ImageNet_dataset(shape=(-1, 256, 256, 3), plotable=False):