Skip to main content

Module tensorboard_integration

Module tensorboard_integration 

Source
Expand description

TensorBoard integration for exporting training metrics and visualizations

This module provides functionality to export TrustformeRS debugging data to TensorBoard format, enabling integration with TensorBoard’s powerful visualization tools.

Structs§

EmbeddingEvent
Embedding event for projector visualization
GraphDef
Graph definition for model structure
GraphNode
Graph node for model architecture visualization
HistogramEvent
Histogram event for logging distributions
ScalarEvent
Scalar event for logging single values
TensorBoardWriter
TensorBoard event writer for logging scalars, histograms, and embeddings
TextEvent
Text event for logging text data

Functions§

create_graph_node
Helper function to create a graph node
tensor_to_histogram_values
Helper to convert tensor statistics to histogram-compatible format