home > examples > optimization

Saving to tensorboard

This directory contains examples on how to visualize data in tensorboard with Objax.

python3 examples/jaxboard/summary.py
tensorboard --logdir experiments/summary_test/tb