wip burn-rl dqn example

This commit is contained in:
Henri Bourcereau 2025-07-08 21:58:15 +02:00
parent b98a135749
commit 354dcfd341
10 changed files with 224 additions and 20 deletions

16
bot/src/burnrl/main.rs Normal file
View 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);
}