From 3ee56661b2369faa8497cf84dc50491f8be24b33 Mon Sep 17 00:00:00 2001 From: wukesong Date: Mon, 22 Jun 2020 19:44:52 +0800 Subject: [PATCH] modify dataset.py --- model_zoo/alexnet/src/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/alexnet/src/dataset.py b/model_zoo/alexnet/src/dataset.py index fe1822579..6e9f310be 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"): -- GitLab