Add initial ML implementation
Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>
This commit is contained in:
13
Cargo.toml
13
Cargo.toml
@@ -11,6 +11,10 @@ path = "src/lib/qrust/mod.rs"
|
||||
name = "qrust"
|
||||
path = "src/bin/qrust/mod.rs"
|
||||
|
||||
[[bin]]
|
||||
name = "trainer"
|
||||
path = "src/bin/trainer/mod.rs"
|
||||
|
||||
[profile.release]
|
||||
panic = 'abort'
|
||||
strip = true
|
||||
@@ -70,7 +74,14 @@ lazy_static = "1.4.0"
|
||||
nonempty = { version = "0.10.0", features = [
|
||||
"serialize",
|
||||
] }
|
||||
rand = "0.8.5"
|
||||
rayon = "1.9.0"
|
||||
burn = { version = "0.12.1", features = [
|
||||
"wgpu",
|
||||
"cuda",
|
||||
"tui",
|
||||
"train",
|
||||
] }
|
||||
|
||||
[dev-dependencies]
|
||||
claims = "0.7.1"
|
||||
serde_test = "1.0.176"
|
||||
|
Reference in New Issue
Block a user