tweak model

This commit is contained in:
2026-03-27 16:04:03 -07:00
parent 1e7bb6c4ce
commit 03f5eb5c13

View File

@@ -6,9 +6,9 @@ pub type InferenceBackend = burn::backend::Cuda<f32>;
pub type TrainingBackend = Autodiff<InferenceBackend>;
pub const SIZE: glam::UVec2 = glam::uvec2(64, 36);
pub const POSITION_HISTORY: usize = 4;
pub const POSITION_HISTORY: usize = 10;
pub const INPUT: usize = (SIZE.x * SIZE.y) as usize + POSITION_HISTORY * 3;
pub const HIDDEN: [usize; 2] = [INPUT >> 3, INPUT >> 7];
pub const HIDDEN: [usize; 3] = [INPUT >> 3, INPUT >> 5, INPUT >> 7];
// MoveForward
// MoveLeft
// MoveBack