size adjust

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,