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"):