26 lines
629 B
TOML
26 lines
629 B
TOML
[package]
|
|
name = "bot"
|
|
version = "0.1.0"
|
|
edition = "2021"
|
|
|
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
|
|
|
[[bin]]
|
|
name = "train_dqn_burn"
|
|
path = "src/dqn/burnrl/main.rs"
|
|
|
|
[[bin]]
|
|
name = "train_dqn"
|
|
path = "src/bin/train_dqn.rs"
|
|
|
|
[dependencies]
|
|
pretty_assertions = "1.4.0"
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
store = { path = "../store" }
|
|
rand = "0.8"
|
|
env_logger = "0.10"
|
|
burn = { version = "0.17", features = ["ndarray", "autodiff"] }
|
|
burn-rl = { git = "https://github.com/yunjhongwu/burn-rl-examples.git", package = "burn-rl" }
|
|
log = "0.4.20"
|