Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.estimator.SessionRunArgs

View source on GitHub

Represents arguments to be added to a Session.run() call.

@staticmethod
tf.estimator.SessionRunArgs(
    cls, fetches, feed_dict=None, options=None
)

Args:

  • fetches: Exactly like the 'fetches' argument to Session.Run(). Can be a single tensor or op, a list of 'fetches' or a dictionary of fetches. For example: fetches = global_step_tensor fetches = [train_op, summary_op, global_step_tensor] fetches = {'step': global_step_tensor, 'summ': summary_op} Note that this can recurse as expected: fetches = {'step': global_step_tensor, 'ops': [train_op, check_nan_op]}
  • feed_dict: Exactly like the feed_dict argument to Session.Run()
  • options: Exactly like the options argument to Session.run(), i.e., a config_pb2.RunOptions proto.

Attributes:

  • fetches
  • feed_dict
  • options