def _listen_tf_v2_summary(self): with warnings.catch_warnings(): warnings.simplefilter("ignore", Warning) from tensorboard.plugins.scalar import summary_v2 self.log.debug( "wrapping tensorboard.plugins.scalar.summary_v2.scalar") python_util.listen_function(summary_v2, "scalar", self._handle_scalar)
def _listen_tf_v2_summary(self): with warnings.catch_warnings(): warnings.simplefilter("ignore", Warning) # pylint: disable=import-error,no-name-in-module from tensorflow.python.ops import summary_ops_v2 self.log.debug("wrapping tensorflow.python.ops summary_ops_v2.scalar") python_util.listen_function(summary_ops_v2, "scalar", self._handle_scalar_ops_v2)
def _listen_tf_summary(self): # pylint: disable=import-error,no-name-in-module from tensorflow import summary self.log.debug("wrapping tensorflow.summary.scalar") python_util.listen_function(summary, "scalar", self._handle_scalar)
def _listen_tf_v2_summary(self): from tensorboard.plugins.scalar import summary_v2 self.log.debug("wrapping tensorboard.plugins.scalar.summary_v2.scalar") python_util.listen_function(summary_v2, "scalar", self._handle_scalar)
def _patch(): from guild import op from guild import python_util python_util.listen_function(op, "run", _patched_op_run)
def _listen_tf_summary(self): from tensorflow import summary self.log.debug("wrapping tensorflow.summary.scalar") python_util.listen_function(summary, "scalar", self._handle_scalar)