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

misc fixes

Browse Source
This commit is contained in:
drake
2026-08-24 23:19:51 -04:00
parent c40a9667bd
commit 33dc15a597
11 changed files with 1742 additions and 1262 deletions
Download Patch File Download Diff File Expand all files Collapse all files
engine/src/main.rs
+4 -10
View File
@@ -1,9 +1,9 @@
#![recursion_limit = "256"]
use burn::backend::{Autodiff, Cuda};
use burn::optim::AdamConfig;
use burn::tensor::{Device, DeviceIndex};
use engine::mcts::MctsConfig;
use engine::training::train::{train, TrainingConfig};
use engine::train::{train, TrainingConfig};
// fn main() {
// type MyBackend = Wgpu<f32, i32>;
//
@@ -14,13 +14,7 @@ use engine::training::train::{train, TrainingConfig};
// }
fn main() {
// type MyBackend = Wgpu<f32, i32>;
type MyBackend = Cuda<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::cuda::CudaDevice::default();
let device = Device::cuda(DeviceIndex::default());
let mcts_config = MctsConfig::new(400, 1.0, 0.05, 0.25, 128);
@@ -43,7 +37,7 @@ fn main() {
seed: None,
};
train::<MyAutodiffBackend>(training_config, device);
train(training_config, &device);
// let mut board_state: Option<BoardState> = Some(BoardState::new(
// Board::default(),