fix generatorDataset get_dataset_size

pull/1659/head
yanghaitao 5 years ago
parent ad279e90fd
commit 7f5db74457

@ -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