Use TensorBoard to visualize the model and its training progression
View the Model Graph
Visualizing topology graphs in TensorBoard does not require performing any Habana-specific steps.
To save and display a model graph, see the example in TensorBoard.
Set the following environment variables to generate a dump of the training graph:
$ export GRAPH_VISUALIZATION=1
$ # Train your model as usual
Synapse graphs will be written to .graph_dumps directory as *.pbtx files.