TensorBoard: Difference between revisions
No edit summary |
|||
| Line 1: | Line 1: | ||
TensorBoard is a way to visualize your model and various statistics during or after training. | |||
==Custom Usage== | ==Custom Usage== | ||
| Line 18: | Line 19: | ||
==Resources== | ==Resources== | ||
* [https://www.tensorflow.org/tensorboard/get_started Getting started with TensorBoard] | * [https://www.tensorflow.org/tensorboard/get_started Getting started with TensorBoard] | ||
* [https://www.youtube.com/watch?v=eBbEDRsCmv4 Hands-on TensorBoard (TensorFlow Dev Summit 2017)] | |||
Revision as of 13:15, 18 June 2020
TensorBoard is a way to visualize your model and various statistics during or after training.
Custom Usage
If you're using a custom training loop (i.e. gradient tape), then you'll need to set everything up manually.
First create a SummaryWriter
train_log_dir = os.path.join(args.checkpoint_dir, "logs", "train")
train_summary_writer = tf.summary.create_file_writer(train_log_dir)
Scalars
Add scalars using tf.summary.scalar:
with train_summary_writer.as_default():
tf.summary.scalar("training_loss", m_loss.numpy(), step=int(ckpt.step))