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

!422 dataset: repeair GeneratorDataset to_device problem

Browse Source 
Merge pull request !422 from ms_yan/Generator_distri
pull/422/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
parent 7036c4ba3d b9742de3a3
commit f8a66ca4fc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
mindspore/dataset/engine/datasets.py
Unescape View File

@ -599,9 +599,9 @@ class Dataset:
def get_distribution(output_dataset): def get_distribution(output_dataset):
dev_id = 0 dev_id = 0
if isinstance(output_dataset, (StorageDataset, GeneratorDataset, MindDataset)): if isinstance(output_dataset, (StorageDataset, MindDataset)):
return output_dataset.distribution, dev_id return output_dataset.distribution, dev_id
if isinstance(output_dataset, (Cifar10Dataset, Cifar100Dataset, ImageFolderDatasetV2, if isinstance(output_dataset, (Cifar10Dataset, Cifar100Dataset, GeneratorDataset, ImageFolderDatasetV2,
ManifestDataset, MnistDataset, VOCDataset, CelebADataset)): ManifestDataset, MnistDataset, VOCDataset, CelebADataset)):
sampler = output_dataset.sampler sampler = output_dataset.sampler
if isinstance(sampler, samplers.DistributedSampler): if isinstance(sampler, samplers.DistributedSampler):

Write Preview
Loading…
Cancel
Save