Python tensorflow.python.training.basic_session_run_hooks.SessionRunArgs() Examples

The following are 25 code examples of tensorflow.python.training.basic_session_run_hooks.SessionRunArgs(). You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may also want to check out all available functions/classes of the module tensorflow.python.training.basic_session_run_hooks , or try the search function .
Example #1
Source File: cifar10_utils.py    From uai-sdk with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #2
Source File: cifar_utils.py    From class-balanced-loss with MIT License 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #3
Source File: cifar10_utils.py    From g-tensorflow-models with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #4
Source File: tpu_estimator.py    From xlnet with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):
    return basic_session_run_hooks.SessionRunArgs(self._tensors)
Example #5
Source File: tpu_estimator.py    From xlnet with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):
    return session_run_hook.SessionRunArgs(self._scalar_stopping_signal)
Example #6
Source File: tpu_estimator.py    From xlnet with Apache License 2.0 5 votes vote down vote up 
def after_run(self, run_context, run_values):
    # Global step cannot be retrieved via SessionRunArgs and before_run due to
    # race condition.
    global_step = run_context.session.run(self._global_step_tensor)
    if global_step >= self._last_step:
      run_context.request_stop()
    else:
      iterations = self._next_iterations(global_step, self._last_step)
      self._iterations_per_loop_var.load(
          iterations, session=run_context.session)
Example #7
Source File: cifar10_utils.py    From object_detection_with_tensorflow with MIT License 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #8
Source File: cifar10_utils.py    From dragonfly with MIT License 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #9
Source File: cifar10_utils.py    From object_detection_kitti with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #10
Source File: cifar10_utils.py    From Live-feed-object-device-identification-using-Tensorflow-and-OpenCV with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #11
Source File: cifar10_utils.py    From nasbot with MIT License 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #12
Source File: utils.py    From DistributedDeepLearning with MIT License 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
        return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #13
Source File: imagenet_utils.py    From uai-sdk with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #14
Source File: tpu_estimator.py    From Chinese-XLNet with Apache License 2.0 5 votes vote down vote up 
def after_run(self, run_context, run_values):
    # Global step cannot be retrieved via SessionRunArgs and before_run due to
    # race condition.
    global_step = run_context.session.run(self._global_step_tensor)
    if global_step >= self._last_step:
      run_context.request_stop()
    else:
      iterations = self._next_iterations(global_step, self._last_step)
      self._iterations_per_loop_var.load(
          iterations, session=run_context.session)
Example #15
Source File: imagenet_utils.py    From uai-sdk with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #16
Source File: tpu_estimator.py    From transformer-xl with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):
    return basic_session_run_hooks.SessionRunArgs(self._tensors)
Example #17
Source File: tpu_estimator.py    From transformer-xl with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):
    return session_run_hook.SessionRunArgs(self._scalar_stopping_signal)
Example #18
Source File: tpu_estimator.py    From transformer-xl with Apache License 2.0 5 votes vote down vote up 
def after_run(self, run_context, run_values):
    # Global step cannot be retrieved via SessionRunArgs and before_run due to
    # race condition.
    global_step = run_context.session.run(self._global_step_tensor)
    if global_step >= self._last_step:
      run_context.request_stop()
    else:
      iterations = self._next_iterations(global_step, self._last_step)
      self._iterations_per_loop_var.load(
          iterations, session=run_context.session)
Example #19
Source File: tpu_estimator.py    From embedding-as-service with MIT License 5 votes vote down vote up 
def before_run(self, run_context):
    return basic_session_run_hooks.SessionRunArgs(self._tensors)
Example #20
Source File: tpu_estimator.py    From embedding-as-service with MIT License 5 votes vote down vote up 
def before_run(self, run_context):
    return session_run_hook.SessionRunArgs(self._scalar_stopping_signal)
Example #21
Source File: tpu_estimator.py    From embedding-as-service with MIT License 5 votes vote down vote up 
def after_run(self, run_context, run_values):
    # Global step cannot be retrieved via SessionRunArgs and before_run due to
    # race condition.
    global_step = run_context.session.run(self._global_step_tensor)
    if global_step >= self._last_step:
      run_context.request_stop()
    else:
      iterations = self._next_iterations(global_step, self._last_step)
      self._iterations_per_loop_var.load(
          iterations, session=run_context.session)
Example #22
Source File: cifar10_utils.py    From Gun-Detector with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #23
Source File: cifar10_utils.py    From yolo_v2 with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):  # pylint: disable=unused-argument
    return basic_session_run_hooks.SessionRunArgs(self._global_step_tensor)
Example #24
Source File: tpu_estimator.py    From Chinese-XLNet with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):
    return basic_session_run_hooks.SessionRunArgs(self._tensors)
Example #25
Source File: tpu_estimator.py    From Chinese-XLNet with Apache License 2.0 5 votes vote down vote up 
def before_run(self, run_context):
    return session_run_hook.SessionRunArgs(self._scalar_stopping_signal)