3个版本
0.5.7 | 2022年3月12日 |
---|---|
0.5.6 | 2022年3月7日 |
0.5.5 | 2022年3月7日 |
#352 in 机器学习
264 下载量/月
在 5 个crate中使用(通过 tensorboard-rs)
765KB
16K SLoC
在Rust中写入tensorboard
在Rust中写入TensorBoard事件。
- 可以写入
scalar
、image
、histogram
。
示例
- 在一个图中写入多个标量。
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();
依赖关系
~1.5MB
~28K SLoC