clankers-cli 0.1.3

Command-line interface for clankeRS
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

use anyhow::Result;
use clankers_ml::ModelValidator;

pub fn execute(
    _pytorch: Option<&str>,
    _checkpoint: Option<&str>,
    onnx: &str,
    samples: &str,
    tolerance: f32,
) -> Result<()> {
    let report = ModelValidator::new()
        .onnx_model(onnx)
        .sample_inputs(samples)
        .tolerance(tolerance)
        .validate()
        .map_err(|e| anyhow::anyhow!("{e}"))?;

    report.print();

    if !report.passed {
        anyhow::bail!("model validation failed");
    }
    Ok(())
}