pipehappy1 / tensorboard-rs

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Not active, checkout https://github.com/pipehappy1/auto-diff

Write to tensorboard in Rust

Write TensorBoard events in Rust.

  • Can write scalar, image, histogram.

Example

  • Write multiple scalar in one plot.
	let mut writer = SummaryWriter::new(&("./logdir".to_string()));

    for n_iter in 0..100 {
        let mut map = HashMap::new();
        map.insert("x1".to_string(), (n_iter as f32));
        map.insert("x^2".to_string(), (n_iter as f32) * (n_iter as f32));
        writer.add_scalars("data/scalar_group", &map, n_iter);
    }
	writer.flush();

About

License:Other


Languages

Language:Rust 99.8%Language:Shell 0.2%