Struct LargeModelVisualizer 

pub struct LargeModelVisualizer { /* private fields */ }

Large model visualizer with memory-efficient rendering

Features:

• Smart layer sampling (visualize representative layers)
• Hierarchical graph rendering (collapse/expand sections)
• Streaming visualization (process in chunks)
• Memory-bounded caching
• Progressive loading

Implementations

impl LargeModelVisualizer

pub fn new(config: LargeModelVisualizerConfig) -> Self

Create a new large model visualizer

Arguments

• config - Visualizer configuration

Example

use trustformers_debug::{LargeModelVisualizer, LargeModelVisualizerConfig};

let config = LargeModelVisualizerConfig::default();
let visualizer = LargeModelVisualizer::new(config);

pub fn add_layer(&self, metadata: LayerMetadata) -> Result<()>

Add layer metadata to the visualizer

Arguments

• metadata - Layer metadata

pub fn determine_sampling(&self) -> Result<Vec<usize>>

Analyze model and determine sampling strategy

Returns

Indices of layers to visualize in detail

pub fn create_layer_groups(&self) -> Result<Vec<LayerGroup>>

Create hierarchical layer groups

Groups layers by type or sequential blocks for collapsible visualization

pub fn visualize( &self, output_path: Option<String>, ) -> Result<VisualizationResult>

Generate visualization with memory-efficient rendering

Arguments

• output_path - Optional output file path

Returns

Visualization result with statistics

pub fn get_progress(&self) -> f64

Get current visualization progress (0.0-1.0)

pub fn get_memory_stats(&self) -> MemoryStats

Get memory usage statistics

Trait Implementations

impl Debug for LargeModelVisualizer

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

Formats the value using the given formatter.