提交 cab2612c 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!1662 fix get_dataset_size error for GeneratorDataset

Merge pull request !1662 from yanghaitao/yht_generator_get_dataset_size_r0.3
......@@ -2951,7 +2951,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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册