speed up optim

2026-08-25 13:46:00 -04:00
parent e327de00a3
commit d3bc5fbd53
4 changed files with 98 additions and 103 deletions
@@ -16,7 +16,7 @@ use engine::train::{train, TrainingConfig};
 fn main() {
    let device = Device::cuda(DeviceIndex::default());

-    let mcts_config = MctsConfig::new(100, 1.0, 0.05, 0.25, 128);
+    let mcts_config = MctsConfig::new(150, 1.0, 0.05, 0.25, 128);

    let adam_config = AdamConfig::new();