Browse Source

!2459 modify dataset.py

Merge pull request !2459 from wukesong/mkdify_alexnet
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
d1f107c89a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/alexnet/src/dataset.py

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


Loading…
Cancel
Save