runner.DatasetProvider

December 14, 2023 ยท View on GitHub

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.