DatasetGenerator
class DatasetGenerator
__init__
def __init__(batch_size, shuffle_buffer_size, rank)
__call__
def __call__(data_generator_func)
Initializes a dataset generator.
Args
- data_generator_func: Callable that returns an iterable over the data to be batched, e.g. lambda