foxhunt/inspect_safetensors.rs

use candle_core::{Device, safetensors};
use std::collections::HashMap;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let model_path = "/tmp/dqn_final_model.safetensors";

    // Load SafeTensors
    let device = Device::Cpu;
    let tensors: HashMap<String, candle_core::Tensor> = safetensors::load(model_path, &device)?;

    println!("=== DQN Model Tensor Structure ===");
    println!("Total tensors: {}", tensors.len());
    println!("\nTensor names and shapes:");

    let mut names: Vec<_> = tensors.keys().collect();
    names.sort();

    for name in names {
        if let Some(tensor) = tensors.get(name) {
            let shape = tensor.shape();
            println!("  {} {:?}", name, shape.dims());
        }
    }

    Ok(())
}