Struct UnifiedDebugSession 

pub struct UnifiedDebugSession { /* private fields */ }

Unified debugging session that manages multiple debugging tools

Implementations

impl UnifiedDebugSession

pub fn new<P: AsRef<Path>>(output_dir: P) -> Result<Self>

Create a new debugging session

Arguments

• output_dir - Directory where debugging outputs will be saved

Example

use trustformers_debug::UnifiedDebugSession;

let session = UnifiedDebugSession::new("debug_outputs").unwrap();

pub fn with_config<P: AsRef<Path>>( output_dir: P, config: UnifiedDebugSessionConfig, ) -> Result<Self>

Create a debugging session with custom configuration

pub fn log_scalar(&mut self, tag: &str, value: f64) -> Result<()>

Log a scalar metric

pub fn log_scalars(&mut self, tag: &str, values: &[(&str, f64)]) -> Result<()>

Log multiple scalars at once

pub fn log_histogram(&mut self, tag: &str, values: &[f64]) -> Result<()>

Log histogram (e.g., weight distribution)

pub fn register_activations( &mut self, layer_name: &str, values: Vec<f32>, shape: Vec<usize>, ) -> Result<()>

Register layer activations

pub fn register_attention( &mut self, layer_name: &str, weights: Vec<Vec<Vec<f64>>>, tokens: Vec<String>, ) -> Result<()>

Register attention weights

pub fn check_stability( &mut self, layer_name: &str, values: &[f64], ) -> Result<usize>

Check tensor for numerical stability

pub fn step(&mut self)

Increment step counter

pub fn current_step(&self) -> u64

Get current step

pub fn save(&mut self) -> Result<()>

Save all accumulated data to disk

pub fn export_activation_viz( &self, layer_name: &str, filename: &str, ) -> Result<()>

Export activation visualization for a specific layer

pub fn export_attention_viz( &self, layer_name: &str, filename: &str, ) -> Result<()>

Export attention visualization for a specific layer

pub fn export_attention_bertviz( &self, layer_name: &str, filename: &str, ) -> Result<()>

Export attention as BertViz HTML

pub fn summary(&self) -> SessionSummary

Get session summary

pub fn session_dir(&self) -> &Path

Get session directory path

pub fn print_summary(&self) -> String

Print summary to string

pub fn close(self) -> Result<SessionSummary>

Close session and perform final save

pub fn activation_visualizer(&self) -> &ActivationVisualizer

Get reference to activation visualizer

pub fn attention_visualizer(&self) -> &AttentionVisualizer

Get reference to attention visualizer

pub fn stability_checker(&self) -> &StabilityChecker

Get reference to stability checker

pub fn graph_visualizer_mut(&mut self) -> Option<&mut GraphVisualizer>

Get mutable reference to graph visualizer (if enabled)

pub fn export_model_netron( &self, model_name: &str, description: &str, ) -> Result<PathBuf>

Export model architecture to Netron format

Trait Implementations

impl Debug for UnifiedDebugSession

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for UnifiedDebugSession

fn drop(&mut self)

Executes the destructor for this type.