提交 d1f107c8 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2459 modify dataset.py

Merge pull request !2459 from wukesong/mkdify_alexnet
......@@ -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"):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册