DatasetProvider.md

runner.DatasetProvider

View source on GitHub

Helper class that provides a standard way to create an ABC using inheritance.

Methods

get_dataset

View source

@abc.abstractmethod
get_dataset(
    context: tf.distribute.InputContext
) -> tf.data.Dataset

Get a tf.data.Dataset by context per replica.