p06452378
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!2459 modify dataset.py 

Merge pull request !2459 from wukesong/mkdify_alexnet
This commit is contained in:
mindspore-ci-bot 2020-06-23 10:51:40 +08:00 committed by Gitee
parent 8e5d8a1cf7 3ee56661b2
commit d1f107c89a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/alexnet/src/dataset.py
View File

@ -16,11 +16,11 @@
Produce the dataset Produce the dataset
""" """
from config import alexnet_cfg as cfg
import mindspore.dataset as ds import mindspore.dataset as ds
import mindspore.dataset.transforms.c_transforms as C import mindspore.dataset.transforms.c_transforms as C
import mindspore.dataset.transforms.vision.c_transforms as CV import mindspore.dataset.transforms.vision.c_transforms as CV
from mindspore.common import dtype as mstype 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"): def create_dataset_mnist(data_path, batch_size=32, repeat_size=1, status="train"):