Browse Source

fix generatorDataset get_dataset_size

tags/v0.5.0-beta
yanghaitao 5 years ago
parent
commit
7f5db74457
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/dataset/engine/datasets.py

+ 2
- 1
mindspore/dataset/engine/datasets.py View File

@@ -2948,7 +2948,8 @@ class GeneratorDataset(MappableDataset):

if rows_from_sampler is None:
return self._dataset_size

if self._dataset_size is None:
return None
return min(rows_from_sampler, self._dataset_size)

# manually set dataset_size as a temporary solution.


Loading…
Cancel
Save