Skip to content

Instantly share code, notes, and snippets.

@Mehanik

Mehanik/keras_to_tensorboard.py

Created December 17, 2019 10:39
Show Gist options
  • Save Mehanik/acd2507f3fae778331bc3701d139e298 to your computer and use it in GitHub Desktop.
Save Mehanik/acd2507f3fae778331bc3701d139e298 to your computer and use it in GitHub Desktop.
Download ZIP
Save keras model for viewing in Tensorboard
Raw
keras_to_tensorboard.py
# tested on Tf 1.15
import tensorflow as tf
m = create_model(...)
m.compile(Adam(lr=1e-3), loss=["binary_crossentropy"])
writer = tf.summary.FileWriter(logdir='logdir', graph=tf.get_default_graph())
writer.flush()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment