tf.contrib.distribute.StandardInputStep

Step with a standard implementation of input handling.

Inherits From: Step

tf.contrib.distribute.StandardInputStep(
    dataset_fn, distribution
)

Args
dataset_fn	a function that returns a tf.data Dataset that produces the input for the model.

Attributes
distribution

Methods

initialize

View source

initialize()

__call__

View source

__call__()

Perform one step of this training algorithm.