Update to burn v0.14.0 and switch to .mpk model file
This commit is contained in:
10
Cargo.toml
10
Cargo.toml
@@ -14,11 +14,11 @@ git = "https://github.com/burn-rs/burn.git"
|
||||
optional = true
|
||||
|
||||
[dependencies]
|
||||
burn = { git = "https://github.com/burn-rs/burn.git" }
|
||||
burn-ndarray = { package = "burn-ndarray", git = "https://github.com/burn-rs/burn.git" }
|
||||
burn-tch = { package = "burn-tch", git = "https://github.com/burn-rs/burn.git" }
|
||||
burn-autodiff = { package = "burn-autodiff", git = "https://github.com/burn-rs/burn.git" }
|
||||
tch = "0.13.0"
|
||||
burn = "0.14.0"
|
||||
burn-ndarray = "0.14.0"
|
||||
burn-tch = "0.14.0"
|
||||
burn-autodiff = "0.14.0"
|
||||
tch = "0.15.0"
|
||||
serde = {version = "1.0.171", features = ["std", "derive"]}
|
||||
npy = "0.4.0"
|
||||
num-traits = "0.2.15"
|
||||
|
||||
Reference in New Issue
Block a user