drake/chess_dragon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixes

Browse Source
This commit is contained in:
drake
2026-05-24 13:28:33 -05:00
parent be589de6f4
commit 23135b4386
3 changed files with 164 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
engine/src/main.rs
+6 -5
View File
@@ -1,7 +1,8 @@
#![recursion_limit = "256"]
use burn::backend::{Autodiff, Wgpu};
use burn::backend::Autodiff;
use burn::optim::AdamConfig;
use burn_ndarray::NdArray;
use engine::mcts::MctsConfig;
use engine::training::train::{train, TrainingConfig};
// fn main() {
@@ -14,12 +15,12 @@ use engine::training::train::{train, TrainingConfig};
// }
fn main() {
type MyBackend = Wgpu<f32, i32>;
// type MyBackend = Wgpu<f32, i32>;
// type MyBackend = Cuda<f32, i32>;
// type MyBackend = NdArray<f32, i32>;
type MyBackend = NdArray<f32, i32>;
type MyAutodiffBackend = Autodiff<MyBackend>;
let device = burn::backend::wgpu::WgpuDevice::default();
// let device = burn::backend::ndarray::NdArrayDevice::default();
// let device = burn::backend::wgpu::WgpuDevice::default();
let device = burn::backend::ndarray::NdArrayDevice::default();
// let device = burn::backend::cuda::CudaDevice::default();
let mcts_config = MctsConfig::new(100, 1.0, 0.05, 0.25);