Browse Source

!1659 fix get_dataset_size error for GeneratorDataset

Merge pull request !1659 from yanghaitao/yht_generator_get_dataset_size
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
58794ab2d9
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

@@ -3011,7 +3011,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