View source on GitHub
|
Class StandardInputStep
Step with a standard implementation of input handling.
Inherits From: Step
Args:
dataset_fn: a function that returns a tf.data Dataset that produces the input for the model.
__init__
__init__(
dataset_fn,
distribution
)
Initialize self. See help(type(self)) for accurate signature.
Properties
distribution
Methods
tf.contrib.distribute.StandardInputStep.__call__
__call__()
Perform one step of this training algorithm.
tf.contrib.distribute.StandardInputStep.initialize
initialize()