From 55dc0bf737cfa37ec9f0e868f37822172c14ecfb Mon Sep 17 00:00:00 2001
From: DragonDuck24 <dragonduckymarino@gmail.com>
Date: Tue, 25 Aug 2026 13:51:55 -0400
Subject: [PATCH] size adjust

---
 engine/src/main.rs  | 10 +++++-----
 engine/src/train.rs |  6 +++---
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/engine/src/main.rs b/engine/src/main.rs
index aedc872..2b05886 100644
--- a/engine/src/main.rs
+++ b/engine/src/main.rs
@@ -16,7 +16,7 @@ use engine::train::{train, TrainingConfig};
 fn main() {
     let device = Device::cuda(DeviceIndex::default());
 
-    let mcts_config = MctsConfig::new(150, 1.0, 0.05, 0.25, 128);
+    let mcts_config = MctsConfig::new(200, 1.0, 0.05, 0.25, 128);
 
     let adam_config = AdamConfig::new();
 
@@ -26,11 +26,11 @@ fn main() {
         max_depth: 100, // unused
         model_name: String::from("Test1"),
         load_model: false,
-        hidden_channels: 64,
+        hidden_channels: 48,
         num_blocks: 4,
-        batch_size: 1024,         // num positions sampled to a batch (2048)
-        num_episodes: 100,        // num games generated per iteration (5000)
-        buffer_max_size: 200_000, // max number of samples in the buffer
+        batch_size: 1024,        // num positions sampled to a batch (2048)
+        num_episodes: 100,       // num games generated per iteration (5000)
+        buffer_max_size: 25_000, // max number of samples in the buffer
         mcts_config,
         optimizer: adam_config,
         lr: 2e-4,
diff --git a/engine/src/train.rs b/engine/src/train.rs
index 61264a6..3b9d86e 100644
--- a/engine/src/train.rs
+++ b/engine/src/train.rs
@@ -119,9 +119,9 @@ pub fn train(training_config: TrainingConfig, device: impl Into<Device>) {
             let mut ply_count: usize = 0;
 
             while board_state.status == BoardStateStatus::Ongoing {
-                let before = Instant::now();
+                // let before = Instant::now();
                 let results = mcts.search(&board_state, &infer_model, &device);
-                println!("{}", before.elapsed().as_millis());
+                // println!("{}", before.elapsed().as_millis());
                 episode_buffer.push(results);
 
                 let temp = if ply_count < 30 { 1.0 } else { 0.0 };
@@ -130,7 +130,7 @@ pub fn train(training_config: TrainingConfig, device: impl Into<Device>) {
 
                 let stm = board_state.board.side_to_move();
                 let mv = sample_move(&adjusted, &mut rng, stm).unwrap();
-                println!("playing move: {}", mv);
+                // println!("playing move: {}", mv);
                 if episode == 0 {
                     game_hist.push_str(mv.to_string().as_str());
                     game_hist.push(' ');