Source code for ASTROMER.core.tboard

import tensorflow as tf


[docs]def draw_graph(model, dataset, writer, logdir=''): '''Decorator that reports store fn graph.''' @tf.function def fn(x): x = model(x) tf.summary.trace_on(graph=True, profiler=False) fn(dataset) with writer.as_default(): tf.summary.trace_export( name='model', step=0, profiler_outdir=logdir)
[docs]def save_scalar(writer, value, step, name=''): with writer.as_default(): tf.summary.scalar(name, value.result(), step=step)