From 03f5eb5c134c805de4446277aa6a278d5398db96 Mon Sep 17 00:00:00 2001
From: Rhys Lloyd <krakow20@gmail.com>
Date: Fri, 27 Mar 2026 16:04:03 -0700
Subject: [PATCH] tweak model

---
 src/net.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/net.rs b/src/net.rs
index a43770d..f427588 100644
--- a/src/net.rs
+++ b/src/net.rs
@@ -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