m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modify dataset.py

Browse Source
pull/2459/head
wukesong 5 years ago
parent 817d1ae2e6
commit 3ee56661b2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
model_zoo/alexnet/src/dataset.py
Unescape 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"):

Write Preview
Loading…
Cancel
Save