def scalar(self, *args, **kwargs):
kwargs.setdefault('step', 1)
writer = tf2.summary.create_file_writer(self.get_temp_dir())
with writer.as_default():
summary.scalar(*args, **kwargs)
writer.close()
return self.read_single_event_from_eventfile().summary
def scalar_event(self, *args, **kwargs):
writer = tf.contrib.summary.create_file_writer(self.get_temp_dir())
with writer.as_default(), tf.contrib.summary.always_record_summaries():
summary.scalar(*args, **kwargs)
return self.read_single_event_from_eventfile()
def graph_fn():
# Recreate the active scope inside the defun since it won't propagate.
with tf.name_scope(scope):
summary.scalar(*args, **kwargs)
def write_scalar_event(self, *args, **kwargs):
kwargs.setdefault("step", 1)
writer = tf2.summary.create_file_writer(self.get_temp_dir())
with writer.as_default():
summary.scalar(*args, **kwargs)
writer.close()