aprender-train-bench 0.41.0

Distillation benchmarking and hyperparameter sweep tool
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

[package]
name = "aprender-train-bench"
version.workspace = true
edition = "2021"
authors = ["PAIML <hello@paiml.com>"]
description = "Distillation benchmarking and hyperparameter sweep tool"
license = "MIT"
repository = "https://github.com/paiml/entrenar"

[lib]
name = "entrenar_bench"
path = "src/lib.rs"

[[bin]]
name = "aprender-train-bench"
path = "src/main.rs"

[lints]
workspace = true

[dependencies]
entrenar = { version = "0.41.0", path = "../aprender-train", package = "aprender-train" }
entrenar-common = { version = "0.41.0", path = "../aprender-train-common", package = "aprender-train-common" }
clap = { version = "4.5", features = ["derive"] }
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
thiserror = "2.0"

[dev-dependencies]
proptest = "1.4"
tempfile = "3.8"
criterion = { workspace = true }
assert_cmd = "2"
predicates = "3"

[[bench]]
name = "sweep_benchmarks"
harness = false