class_docstr: Add timeout option to each sess.run call.
class_docstr_tok: ["Add","timeout","option","to","each","sess.run","call","."]
docstr: Args:
      timeout_in_ms (int):
docstr_tok: ["Args",":","timeout_in_ms","(","int",")",":"]
code: def __init__(self, timeout_in_ms):
      """
      Args:
          timeout_in_ms (int):
      """
      self._timeout = int(timeout_in_ms)
  
      opt = tf.RunOptions(timeout_in_ms=timeout_in_ms)
      self._runargs = tf.train.SessionRunArgs(fetches=[], options=opt)
  
imports: import tqdm
  from six.moves import zip
  from ..compat import tfv1
  from ..tfutils.common import get_global_step_var, get_op_tensor_name
  from ..utils import logger
  from ..utils.naming import GLOBAL_STEP_INCR_OP_NAME
  from ..utils.utils import get_tqdm_kwargs
  from .base import Callback
label_desc: Use image node_id 1 to create a new SessionRunTimeout object from inherited base classes: Callback with example: obj = SessionRunTimeout(timeout_in_ms)