burn/examples/custom-training-loop/Cargo.toml

20 lines
483 B
TOML
Raw Normal View History

2023-09-03 01:09:03 +08:00
[package]
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
edition = "2021"
license = "MIT OR Apache-2.0"
name = "custom-training-loop"
publish = false
version = "0.11.0"
2023-09-03 01:09:03 +08:00
[dependencies]
burn = {path = "../../burn", features=["autodiff", "wgpu"]}
2023-09-03 01:09:03 +08:00
guide = {path = "../guide"}
# Serialization
log = {workspace = true}
serde = {workspace = true, features = ["std", "derive"]}
# Wgpu internal dependencies
derive-new = { workspace = true }
bytemuck = { workspace = true }