wip burn-rl dqn example
This commit is contained in:
parent
b98a135749
commit
354dcfd341
10 changed files with 224 additions and 20 deletions
16
bot/src/burnrl/main.rs
Normal file
16
bot/src/burnrl/main.rs
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
use burn::backend::{Autodiff, NdArray};
|
||||
use burn_rl::base::ElemType;
|
||||
use bot::burnrl::{
|
||||
dqn_model,
|
||||
environment,
|
||||
utils::demo_model,
|
||||
};
|
||||
|
||||
type Backend = Autodiff<NdArray<ElemType>>;
|
||||
type Env = environment::TrictracEnvironment;
|
||||
|
||||
fn main() {
|
||||
let agent = dqn_model::run::<Env, Backend>(512, false); //true);
|
||||
|
||||
demo_model::<Env>(agent);
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue