diff --git a/model_zoo/alexnet/src/dataset.py b/model_zoo/alexnet/src/dataset.py
index fe182257910e46b1956063b0bf6125563b74072c..6e9f310bed61b116673a84b2b6a86376e9964234 100644
--- a/model_zoo/alexnet/src/dataset.py
+++ b/model_zoo/alexnet/src/dataset.py
@@ -16,11 +16,11 @@
 Produce the dataset
 """
 
-from config import alexnet_cfg as cfg
 import mindspore.dataset as ds
 import mindspore.dataset.transforms.c_transforms as C
 import mindspore.dataset.transforms.vision.c_transforms as CV
 from mindspore.common import dtype as mstype
+from .config import alexnet_cfg as cfg
 
 
 def create_dataset_mnist(data_path, batch_size=32, repeat_size=1, status="train"):
diff --git a/model_zoo/lenet/src/lenet.py b/model_zoo/lenet/src/lenet.py
index 3864315dba35f1e4fd013317b71572ad6db82e95..a570c6cd9de2336c6ba7c413d7046da56d6a44f2 100644
--- a/model_zoo/lenet/src/lenet.py
+++ b/model_zoo/lenet/src/lenet.py
@@ -43,11 +43,12 @@ class LeNet5(nn.Cell):
 
     Args:
         num_class (int): Num classes. Default: 10.
+        channel (int): Num channels. Default: 1.
 
     Returns:
         Tensor, output tensor
     Examples:
-        >>> LeNet(num_class=10)
+        >>> LeNet(num_class=10, channel=1)
 
     """
     def __init__(self, num_class=10, channel=1):