henri/trictrac
1
1
Fork 0
Code Issues 13 Pull requests Projects 1 Releases Packages Wiki Activity Actions

wip

Browse source
This commit is contained in:
Henri Bourcereau 2025-07-23 17:25:05 +02:00
parent 354dcfd341
commit c6d33555ec
5 changed files with 25 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
bot/src/burnrl/main.rs
View file

@ -1,16 +1,13 @@
use bot::burnrl::{dqn_model, environment, utils::demo_model};
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);
let num_episodes = 3;
let agent = dqn_model::run::<Env, Backend>(num_episodes, false); //true);
demo_model::<Env>(agent);
}